diff --git a/features/steps/strategize_decision_recording_steps.py b/features/steps/strategize_decision_recording_steps.py index 4a0769881..8dcd831ab 100644 --- a/features/steps/strategize_decision_recording_steps.py +++ b/features/steps/strategize_decision_recording_steps.py @@ -320,7 +320,12 @@ def step_then_decision_phase(context, phase): """ assert context.last_decision is not None # Strategize-phase decision types accepted by the hook - strategize_types = {"strategy_choice", "resource_selection", "subplan_spawn", "invariant_enforced"} + strategize_types = { + "strategy_choice", + "resource_selection", + "subplan_spawn", + "invariant_enforced", + } assert context.last_decision.decision_type.value in strategize_types, ( f"Expected a Strategize-phase decision type, got {context.last_decision.decision_type.value!r}" ) diff --git a/tests/actor/test_registry_builtin_yaml.py b/tests/actor/test_registry_builtin_yaml.py index b7e86e9ec..750823917 100644 --- a/tests/actor/test_registry_builtin_yaml.py +++ b/tests/actor/test_registry_builtin_yaml.py @@ -17,13 +17,12 @@ import yaml from cleveragents.actor.registry import ActorRegistry from cleveragents.actor.schema import ActorConfigSchema, ActorType, is_v3_yaml -from cleveragents.config.settings import ProviderDefaults, Settings +from cleveragents.config.settings import ProviderDefaults from cleveragents.core.exceptions import NotFoundError from cleveragents.domain.models.core.actor import Actor from cleveragents.providers.registry import ( ProviderCapabilities, ProviderInfo, - ProviderRegistry, ProviderType, )