fix: workaround for updating function image URI

Signed-off-by: SdgJlbl <sarah.diot-girard@owkin.com>
Substra · Oct 17, 2023 · 113b1fe · 113b1fe
1 parent 553e85d
commit 113b1fe
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/backend/substrapp/compute_tasks/image_builder.py b/backend/substrapp/compute_tasks/image_builder.py
@@ -43,11 +43,13 @@ def load_remote_function_image(function: orchestrator.Function, channel: str) ->
     container_image_tag = utils.container_image_tag_from_function(function)
     # Ask the backend owner of the function if it's available
 
+    api_function = ApiFunction.objects.get(key=function.key)
+
     function_image_content = organization_client.get(
         channel=channel,
         organization_id=function.owner,
-        url=function.image.uri,
-        checksum=function.image.checksum,
+        url=api_function.image_address,
+        checksum=api_function.image_checksum,
     )
 
     os.makedirs(SUBTUPLE_TMP_DIR, exist_ok=True)