diff --git a/hypothesis-python/src/hypothesis/extra/ghostwriter.py b/hypothesis-python/src/hypothesis/extra/ghostwriter.py index 99f56b3b00..1ea7b760e2 100644 --- a/hypothesis-python/src/hypothesis/extra/ghostwriter.py +++ b/hypothesis-python/src/hypothesis/extra/ghostwriter.py @@ -603,7 +603,8 @@ def _imports_for_strategy(strategy): ): return { imp - for arg in set(strategy._LazyStrategy__args) | set(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."):