Skip to content
Snippets Groups Projects
Commit e00c5049 authored by Manon Blanco's avatar Manon Blanco Committed by Yoann Schneider
Browse files

Apply suggestions

parent a991ded7
No related branches found
No related tags found
1 merge request!500Store model_version details regardless of model version configuration
......@@ -264,7 +264,7 @@ class BaseWorker:
# Load model version configuration when available
model_version = worker_run.get("model_version")
if model_version and model_version.get("configuration"):
if model_version:
logger.info("Loaded model version configuration from WorkerRun")
self.model_configuration.update(model_version.get("configuration"))
......@@ -272,7 +272,7 @@ class BaseWorker:
self.model_version_id = model_version.get("id")
# Set model details as worker attribute
self.model_details = model_version.get("model")
self.model_details = model_version.get("model", {})
# Retrieve initial configuration from API
self.config = worker_version["configuration"].get("configuration", {})
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment