From aeef3a0727e34180b55ad8ef472e4b79152027ec Mon Sep 17 00:00:00 2001 From: andrewtarzia Date: Thu, 21 Mar 2024 11:58:32 +0100 Subject: [PATCH] Fix types. --- .../molecules/building_block/test_with_functional_groups.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/molecular/molecules/building_block/test_with_functional_groups.py b/tests/molecular/molecules/building_block/test_with_functional_groups.py index a5cd557a1..32ccecf31 100644 --- a/tests/molecular/molecules/building_block/test_with_functional_groups.py +++ b/tests/molecular/molecules/building_block/test_with_functional_groups.py @@ -10,7 +10,9 @@ def test_with_functional_groups( building_block: stk.BuildingBlock, - get_functional_groups: abc.Callable[stk.BuildingBlock], + get_functional_groups: abc.Callable[ + [stk.BuildingBlock, ...], tuple[stk.FunctionalGroup, ...] + ], ) -> None: """ Test :meth:`.BuildingBlock.with_functional_groups`.