瀏覽代碼

Fix issue 2319 (#2321)

Jean-Robin 5 月之前
父節點
當前提交
6678003b3c

+ 3 - 0
taipy/core/_orchestrator/_dispatcher/_job_dispatcher.py

@@ -159,4 +159,7 @@ class _JobDispatcher(threading.Thread):
                 _TaipyLogger._get_logger().error(st)
             _JobManagerFactory._build_manager()._set(job)
         else:
+            for output in job.task.output.values():
+                output.track_edit(job_id=job.id)
+                output.unlock_edit()
             job.completed()

+ 1 - 1
taipy/core/_orchestrator/_dispatcher/_task_function_wrapper.py

@@ -65,7 +65,7 @@ class _TaskFunctionWrapper:
                 for res, dn in zip(_results, outputs):
                     try:
                         data_node = data_manager._get(dn.id)
-                        data_node.write(res, job_id=job_id)
+                        data_node._write(res)
                     except Exception as e:
                         logger.error("Error during write", exc_info=1)
                         exceptions.append(DataNodeWritingError(f"Error writing in datanode id {dn.id}: {e}"))

+ 3 - 3
tests/core/job/test_job.py

@@ -72,10 +72,10 @@ def job(task, job_id):
 
 @pytest.fixture
 def replace_in_memory_write_fct():
-    default_write = InMemoryDataNode.write
-    InMemoryDataNode.write = _error
+    default_write = InMemoryDataNode._write
+    InMemoryDataNode._write = _error
     yield
-    InMemoryDataNode.write = default_write
+    InMemoryDataNode._write = default_write
 
 
 def _foo():