From cbad1a145fa14b5e9b8f4d529653fb8986ea3db7 Mon Sep 17 00:00:00 2001 From: JensHeinrich Date: Fri, 20 Jan 2023 08:04:44 +0100 Subject: [PATCH] Reformat --- hypothesis-python/src/hypothesis/extra/ghostwriter.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/hypothesis-python/src/hypothesis/extra/ghostwriter.py b/hypothesis-python/src/hypothesis/extra/ghostwriter.py index 0e44ed075f..c6fa7a13a7 100644 --- a/hypothesis-python/src/hypothesis/extra/ghostwriter.py +++ b/hypothesis-python/src/hypothesis/extra/ghostwriter.py @@ -597,13 +597,14 @@ def _imports_for_strategy(strategy): # If we have a lazy from_type strategy, because unwrapping it gives us an # error or invalid syntax, import that type and we're done. if isinstance(strategy, LazyStrategy): - if ( - strategy.function.__name__ in ( - st.from_type.__name__, st.from_regex.__name__) + if strategy.function.__name__ in ( + st.from_type.__name__, + st.from_regex.__name__, ): return { imp - for arg in strategy._LazyStrategy__args + strategy._LazyStrategy__kwargs.values + for arg in strategy._LazyStrategy__args + + strategy._LazyStrategy__kwargs.values for imp in _imports_for_object(arg) } elif _get_module(strategy.function).startswith("hypothesis.extra."):