From 6cb475ecdcdaa94c363ce17435728740e3b6ea63 Mon Sep 17 00:00:00 2001 From: Czar Echavez Date: Mon, 17 Jul 2023 11:54:54 +0100 Subject: [PATCH] #1595: set default tc and ldc to None for AgentABC and PrimaiteSession + adding a comment for cli load flag --- src/primaite/agents/agent_abc.py | 4 ++-- src/primaite/agents/sb3.py | 4 ++-- src/primaite/cli.py | 5 +++++ 3 files changed, 9 insertions(+), 4 deletions(-) diff --git a/src/primaite/agents/agent_abc.py b/src/primaite/agents/agent_abc.py index e36196a0..515adfd0 100644 --- a/src/primaite/agents/agent_abc.py +++ b/src/primaite/agents/agent_abc.py @@ -47,8 +47,8 @@ class AgentSessionABC(ABC): @abstractmethod def __init__( self, - training_config_path: Optional[Union[str, Path]] = "", - lay_down_config_path: Optional[Union[str, Path]] = "", + training_config_path: Optional[Union[str, Path]] = None, + lay_down_config_path: Optional[Union[str, Path]] = None, session_path: Optional[Union[str, Path]] = None, ): """ diff --git a/src/primaite/agents/sb3.py b/src/primaite/agents/sb3.py index de70bf26..862a0116 100644 --- a/src/primaite/agents/sb3.py +++ b/src/primaite/agents/sb3.py @@ -21,8 +21,8 @@ class SB3Agent(AgentSessionABC): def __init__( self, - training_config_path: Optional[Union[str, Path]] = "", - lay_down_config_path: Optional[Union[str, Path]] = "", + training_config_path: Optional[Union[str, Path]] = None, + lay_down_config_path: Optional[Union[str, Path]] = None, session_path: Optional[Union[str, Path]] = None, ): """ diff --git a/src/primaite/cli.py b/src/primaite/cli.py index 4f8e8f03..adc9cb32 100644 --- a/src/primaite/cli.py +++ b/src/primaite/cli.py @@ -162,6 +162,11 @@ def session(tc: Optional[str] = None, ldc: Optional[str] = None, load: Optional[ ldc: The lay down config file path. Optional. If no value is passed then example default lay down config is used from: ~/primaite/config/example_config/lay_down/lay_down_config_3_doc_very_basic.yaml. + + load: The directory of a previous session. Optional. If no value is passed, then the session + will use the default training config and laydown config. Inversely, if a training config and laydown config + is passed while a session directory is passed, PrimAITE will load the session and ignore the training config + and laydown config. """ from primaite.config.lay_down_config import dos_very_basic_config_path from primaite.config.training_config import main_training_config_path