Skip to content
Snippets Groups Projects

Store model_version details regardless of model version configuration

Merged Manon Blanco requested to merge always-set-model-attributes into master
All threads resolved!
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
@@ -266,7 +266,7 @@ class BaseWorker:
model_version = worker_run.get("model_version")
if model_version:
logger.info("Loaded model version configuration from WorkerRun")
self.model_configuration.update(model_version.get("configuration", {}))
self.model_configuration.update(model_version["configuration"])
# Set model_version ID as worker attribute
self.model_version_id = model_version["id"]
Loading