fix: casting when receiving FunctionProfilingStep (#932)

* fix: casting when receiving `FunctionProfilingStep` Signed-off-by: Guilhem Barthés <guilhem.barthes@owkin.com> * docs: add change fragment Signed-off-by: Guilhem Barthés <guilhem.barthes@owkin.com> --------- Signed-off-by: Guilhem Barthés <guilhem.barthes@owkin.com>
Substra · Jun 14, 2024 · f00affe · f00affe
1 parent 682c4aa
commit f00affe
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/backend/api/events/sync.py b/backend/api/events/sync.py
@@ -1,3 +1,4 @@
+import datetime
 import secrets
 import string
 from typing import Optional
@@ -434,8 +435,9 @@ def _create_failure_report(data: dict) -> None:
 
 
 def _on_create_profiling_step_event(data: dict) -> None:
+    duration = datetime.timedelta(seconds=data["profiling_step"]["duration"])
     profiling_step = FunctionProfilingStep(
-        function_key=data["asset_key"], duration=data["profiling_step"]["duration"], step=data["profiling_step"]["step"]
+        function_id=data["asset_key"], duration=duration, step=data["profiling_step"]["step"]
     )
     profiling_step.save()
 

diff --git a/changes/932.fixed b/changes/932.fixed
@@ -0,0 +1 @@
+casting when receiving `FunctionProfilingStep` from the orchestrator