From d2eae87a997f8baa3f7aefeefa1f8346620930ab Mon Sep 17 00:00:00 2001
From: manonBlanco <blanco@teklia.com>
Date: Wed, 28 Feb 2024 16:33:35 +0100
Subject: [PATCH] Configuration is always set
---
arkindex_worker/worker/base.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/arkindex_worker/worker/base.py b/arkindex_worker/worker/base.py
index 37da3a56..1be5df4f 100644
--- a/arkindex_worker/worker/base.py
+++ b/arkindex_worker/worker/base.py
@@ -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"]
--
GitLab