From d40463743c3d48d5ae8589e1d8ea5bd68e8127a3 Mon Sep 17 00:00:00 2001 From: andrewtarzia Date: Tue, 23 Jan 2024 15:41:48 +0100 Subject: [PATCH] Fix. --- src/bbprep/_internal/generators/geometry_scanner.py | 2 +- tests/torsion_test/conftest.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/bbprep/_internal/generators/geometry_scanner.py b/src/bbprep/_internal/generators/geometry_scanner.py index c2b90f9..843a7c0 100644 --- 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 index 2332370..06d0a6f 100644 --- 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), ), ), ),