def load_torch_model_training_from_task_id(model_cls: Type[TorchModelTraining], task_id: str) -> TorchModelTraining: task_dir = get_task_dir(model_cls, task_id) if not os.path.exists(task_dir): task_dir = get_interaction_dir(model_cls, task_id) return load_torch_model_training_from_task_dir(model_cls, task_dir)
def output(self): return luigi.LocalTarget(get_task_dir(self.__class__, self.task_id))