diff --git a/src/mednet/scripts/upload.py b/src/mednet/scripts/upload.py index 0b4f01e74dd97befdc6f7a48694b0e125e1cbf60..8d0b14d98bb5439359831c7d127fd89904cab5a1 100644 --- a/src/mednet/scripts/upload.py +++ b/src/mednet/scripts/upload.py @@ -20,7 +20,7 @@ from .click import ConfigCommand logger = setup(__name__.split(".")[0], format="%(levelname)s: %(message)s") -def get_config(): +def _get_gitlab_config(): """Return an instance of the Gitlab object for remote operations. Returns @@ -42,7 +42,7 @@ def get_config(): return gl, config -def create_temp_copy(source, target): +def _create_temp_copy(source, target): """Create a copy of original file in temp folder. Parameters @@ -122,7 +122,7 @@ def upload( """Upload results from an experiment output folder.""" logger.info("Getting Gitlab credentials for accessing to MLFlow server...") - gitlab, config = get_config() + gitlab, config = _get_gitlab_config() project = gitlab.projects.get("biosignal/software/mednet") os.environ["MLFLOW_TRACKING_TOKEN"] = config["idiap"]["private_token"] os.environ["MLFLOW_TRACKING_URI"] = ( @@ -135,7 +135,7 @@ def upload( train_log_file = train_output_folder / "trainlog.pdf" train_model_file = [f for f in train_output_folder.glob("*lowest*")][0] train_model_temp_file = train_model_file.parts[-1].replace("=", "_") - train_model_temp_file = create_temp_copy( + train_model_temp_file = _create_temp_copy( train_model_file, train_model_temp_file ) with train_meta_file.open("r") as f: