test: use standardState in test_workflow

AutoResearch · Nov 30, 2023 · aac4101 · aac4101
1 parent 7360b8e
commit aac4101
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_workflow.py b/tests/test_workflow.py
@@ -8,15 +8,15 @@
 from hypothesis import strategies as st
 
 from autora.serializer import SupportedSerializer, load_state
-from autora.state import StandardState, State
+from autora.state import StandardState
 from autora.workflow.__main__ import main
 
 _logger = logging.getLogger(__name__)
 
 example_workflow_library_module = st.sampled_from(["_example_workflow_library"])
 
 
-def validate_model(state: Optional[State]):
+def validate_model(state: Optional[StandardState]):
     assert state is not None
 
     assert state.conditions is not None