@@ -91,8 +91,6 @@ def __init__(
9191 root_dir : _PATH | None = None ,
9292 teamspace : str | None = None ,
9393 metadata : Dict [str , str ] | None = None ,
94- store_step : bool | None = False ,
95- store_created_at : bool | None = False ,
9694 log_model : bool = False ,
9795 save_logs : bool = False ,
9896 checkpoint_name : str | None = None ,
@@ -144,8 +142,6 @@ def training_step(self, batch, batch_idx: int):
144142 self ._fs = get_filesystem (root_dir )
145143 self ._step = - 1
146144 self ._metadata = metadata or {}
147- self ._store_step = store_step
148- self ._store_created_at = store_created_at
149145 self ._is_ready = False
150146 self ._log_model = log_model
151147 self ._save_logs = save_logs
@@ -220,8 +216,8 @@ def experiment(self) -> Experiment | None:
220216 version = self .version ,
221217 teamspace = self ._teamspace ,
222218 metadata = {k : str (v ) for k , v in self ._metadata .items ()},
223- store_step = self . _store_step ,
224- store_created_at = self . _store_created_at ,
219+ store_step = True ,
220+ store_created_at = True ,
225221 log_dir = self .log_dir ,
226222 save_logs = self ._save_logs ,
227223 )
0 commit comments