Skip to content

Commit f6cf802

Browse files
committed
fixup! train: Return MLflow tracking information
Signed-off-by: Phoevos Kalemkeris <[email protected]>
1 parent fb1d335 commit f6cf802

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

app/trainers/base.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -35,17 +35,17 @@ def __init__(self, config: Settings, model_name: str) -> None:
3535
def model_name(self) -> str:
3636
return self._model_name
3737

38+
@model_name.setter
39+
def model_name(self, model_name: str) -> None:
40+
self._model_name = model_name
41+
3842
@property
3943
def experiment_id(self) -> str:
40-
return self._experiment_id
44+
return self._experiment_id or ""
4145

4246
@property
4347
def run_id(self) -> str:
44-
return self._run_id
45-
46-
@model_name.setter
47-
def model_name(self, model_name: str) -> None:
48-
self._model_name = model_name
48+
return self._run_id or ""
4949

5050
@final
5151
def start_training(self,

0 commit comments

Comments
 (0)