Skip to content

Commit

Permalink
fixup! train: Return MLflow tracking information
Browse files Browse the repository at this point in the history
Signed-off-by: Phoevos Kalemkeris <[email protected]>
  • Loading branch information
phoevos committed Dec 18, 2024
1 parent fb1d335 commit f6cf802
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions app/trainers/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,17 +35,17 @@ def __init__(self, config: Settings, model_name: str) -> None:
def model_name(self) -> str:
return self._model_name

@model_name.setter
def model_name(self, model_name: str) -> None:
self._model_name = model_name

@property
def experiment_id(self) -> str:
return self._experiment_id
return self._experiment_id or ""

@property
def run_id(self) -> str:
return self._run_id

@model_name.setter
def model_name(self, model_name: str) -> None:
self._model_name = model_name
return self._run_id or ""

@final
def start_training(self,
Expand Down

0 comments on commit f6cf802

Please sign in to comment.