Fix import finding

Use function call instead of function Switch to using sets
HypothesisWorks · Feb 16, 2023 · 38ee9d3 · 38ee9d3
1 parent 99490c1
commit 38ee9d3
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/hypothesis-python/src/hypothesis/extra/ghostwriter.py b/hypothesis-python/src/hypothesis/extra/ghostwriter.py
@@ -603,8 +603,7 @@ def _imports_for_strategy(strategy):
         ):
             return {
                 imp
-                for arg in strategy._LazyStrategy__args
-                + strategy._LazyStrategy__kwargs.values
+                for arg in set(strategy._LazyStrategy__args) | set(strategy._LazyStrategy__kwargs.values())
                 for imp in _imports_for_object(arg)
             }
         elif _get_module(strategy.function).startswith("hypothesis.extra."):