Fix.

andrewtarzia · Jan 23, 2024 · d404637 · d404637
1 parent cc7cb00
commit d404637
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/bbprep/_internal/generators/geometry_scanner.py b/src/bbprep/_internal/generators/geometry_scanner.py
@@ -18,7 +18,7 @@ def __init__(
         target_ranges: BondRange
         | AngleRange
         | TorsionRange
-        | tuple[BondRange, AngleRange, TorsionRange],
+        | tuple[BondRange | AngleRange | TorsionRange],
     ) -> None:
         """Initialise generator."""
         if not isinstance(target_ranges, tuple):

diff --git a/tests/torsion_test/conftest.py b/tests/torsion_test/conftest.py
@@ -45,8 +45,8 @@
                     bbprep.generators.TorsionRange(
                         smarts="[#7][#6][#6][#7]",
                         expected_num_atoms=4,
-                        torsion_ids=(0, 1, 2, 3),
-                        torsion_range=range(0, 362, 40),
+                        scanned_ids=(0, 1, 2, 3),
+                        scanned_range=range(0, 362, 40),
                     ),
                 ),
             ),