Set job id on dataset for discovered outputs

This commit is contained in:
mvdbeek
2021-10-26 17:59:04 +02:00
parent a4033ea480
commit 1a6a4b50de
2 changed files with 3 additions and 1 deletions
+1 -1
View File
@@ -1709,7 +1709,7 @@ class JobWrapper(HasResourceParameters):
user=job.user,
tag_handler=self.app.tag_handler.create_tag_handler_session(),
)
import_model_store.perform_import(history=job.history)
import_model_store.perform_import(history=job.history, job=job)
except Exception:
log.exception(f"problem importing job outputs. stdout [{job.stdout}] stderr [{job.stderr}]")
raise
+2
View File
@@ -217,6 +217,8 @@ class ModelImportStore(metaclass=abc.ABCMeta):
if 'id' in dataset_attrs["dataset"] and self.import_options.allow_edit:
dataset_instance.dataset.id = dataset_attrs["dataset"]['id']
if job:
dataset_instance.dataset.job_id = job.id
if 'id' in dataset_attrs and self.import_options.allow_edit and not self.sessionless:
dataset_instance = self.sa_session.query(getattr(model, dataset_attrs['model_class'])).get(dataset_attrs["id"])