98fc1e4c71
- renamed _config_values_main to training_config.py and renamed the ConfigValuesMain class to TrainingConfig. Moved training_config.py to src/primaite/config/training_config.py - Renamed all training config yaml file keys to make creating an instance of TrainingConfig easier. Moved action_type and num_steps over to the training config. - Decoupled the training config and lay down config. - Refactored main.py so that it can be ran from CLI and can take a training config path and a lay down config path. - refactored all outputs so that they save to the session dir. - Added some necessary setup scripts that handle creating app dirs, fronting example config files to the user, fronting demo notebooks to the user, performing clean-up in between installations etc. - Added functions that attempt to retrieve the file path of users example config files that have been fronted by the primaite setup. - Added logging config and a getLogger function in the top-level init. - Refactored all logs entries logged to use a logger using the primaite logging config. - Added basic typer CLI for doing things like setup, viewing logs, viewing primaite version, running a basic session. - Updated test to use new features and config structures. - Began updating docs. More to do here.
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence.
|
|
import importlib.util
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
from primaite import NOTEBOOKS_DIR, getLogger
|
|
|
|
_LOGGER = getLogger(__name__)
|
|
|
|
|
|
def start_jupyter_session():
|
|
"""
|
|
Starts a new Jupyter notebook session in the app notebooks directory.
|
|
|
|
Currently only works on Windows OS.
|
|
|
|
.. todo:: Figure out how to get this working for Linux and MacOS too.
|
|
"""
|
|
if sys.platform == "win32":
|
|
if importlib.util.find_spec("jupyter") is not None:
|
|
# Jupyter is installed
|
|
working_dir = os.getcwd()
|
|
os.chdir(NOTEBOOKS_DIR)
|
|
subprocess.Popen("jupyter lab")
|
|
os.chdir(working_dir)
|
|
else:
|
|
# Jupyter is not installed
|
|
_LOGGER.error("Cannot start jupyter lab as it is not installed")
|
|
else:
|
|
msg = (
|
|
"Feature currently only supported on Windows OS. For "
|
|
"Linux/MacOS users, run 'cd ~/primaite/notebooks; jupyter "
|
|
"lab' from your Python environment."
|
|
)
|
|
_LOGGER.warning(msg)
|