Integrate comments from review

OpenSourceEconomics · Sep 23, 2024 · 7fc731e · 7fc731e
1 parent 9b7133f
commit 7fc731e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/tests/input_processing/test_process_model.py b/tests/input_processing/test_process_model.py
@@ -18,9 +18,7 @@
     process_model,
 )
 from lcm.mark import StochasticInfo
-from tests.test_models import (
-    get_model_config,
-)
+from tests.test_models import get_model_config
 
 
 @dataclass

diff --git a/tests/test_entry_point.py b/tests/test_entry_point.py
@@ -11,9 +11,7 @@
 from lcm.model_functions import get_utility_and_feasibility_function
 from lcm.state_space import create_state_choice_space
 from tests.test_models import get_model_config
-from tests.test_models.deterministic import (
-    RetirementStatus,
-)
+from tests.test_models.deterministic import RetirementStatus
 from tests.test_models.deterministic import utility as iskhakov_et_al_2017_utility
 from tests.test_models.discrete_deterministic import ConsumptionChoice