#1595: minor fix to cli command

This commit is contained in:
Czar Echavez
2023-07-14 16:04:34 +01:00
parent 7c2ff55da2
commit ea7c1519fe

View File

@@ -151,7 +151,7 @@ def setup(overwrite_existing: bool = True):
@app.command()
def session(tc: Optional[str] = None, ldc: Optional[str] = None):
def session(tc: Optional[str] = None, ldc: Optional[str] = None, load: Optional[str] = None):
"""
Run a PrimAITE session.
@@ -167,6 +167,9 @@ def session(tc: Optional[str] = None, ldc: Optional[str] = None):
from primaite.config.training_config import main_training_config_path
from primaite.main import run
if load is not None:
run(session_path=load)
if not tc:
tc = main_training_config_path()