#1595: set default tc and ldc to None for AgentABC and PrimaiteSession + adding a comment for cli load flag

This commit is contained in:
Czar Echavez
2023-07-17 11:54:54 +01:00
parent 232908ac00
commit 6cb475ecdc
3 changed files with 9 additions and 4 deletions

View File

@@ -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,
):
"""

View File

@@ -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,
):
"""

View File

@@ -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