feat: better args parsing

This commit is contained in:
2026-03-15 20:03:01 -03:00
parent 4c8cbc821d
commit 6a154429e9
3 changed files with 13 additions and 6 deletions

View File

@@ -19,13 +19,16 @@ class Player:
self._state_handler = StateHandler(config.simulation_dir)
def play(self, **flags: Unpack[PlayerFlags]):
if not flags["continuation"]:
continuation = flags.get("continuation", False)
force = flags.get("force", False)
state: StateModel = self._state_handler.get(self.config, force=force)
if not continuation and state is not None:
logger.info(
"Continuation flag is not set. Starting a new simulation and deleting any existing state."
)
self._state_handler.delete()
state = self._state_handler.get(self.config, force=flags["force"])
state = self._state_handler.get(self.config, force=force)
if state is None:
state = StateModel.from_config(self.config)
@@ -37,7 +40,7 @@ class Player:
f"Starting cycle {state.current_cycle + 1} of {self.config.max_cyc}."
)
step_directory = self.config.simulation_dir / f"{state.current_cycle::02d}"
step_directory = self.config.simulation_dir / f"{state.current_cycle:02d}"
if not step_directory.exists():
step_directory.mkdir(parents=True)