Skip to content

Commit 38ee9d3

Browse files
JensHeinrichJensHeinrich
authored andcommitted
Fix import finding
Use function call instead of function Switch to using sets
1 parent 99490c1 commit 38ee9d3

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

hypothesis-python/src/hypothesis/extra/ghostwriter.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -603,8 +603,7 @@ def _imports_for_strategy(strategy):
603603
):
604604
return {
605605
imp
606-
for arg in strategy._LazyStrategy__args
607-
+ strategy._LazyStrategy__kwargs.values
606+
for arg in set(strategy._LazyStrategy__args) | set(strategy._LazyStrategy__kwargs.values())
608607
for imp in _imports_for_object(arg)
609608
}
610609
elif _get_module(strategy.function).startswith("hypothesis.extra."):

0 commit comments

Comments
 (0)