Просмотр исходного кода

added enterprise check for task fct wrapper in dev job dispatcher

Toan Quach 7 месяцев назад
Родитель
Сommit
3c5fa46445
1 измененных файлов с 10 добавлено и 1 удалено
  1. 10 1
      taipy/core/_orchestrator/_dispatcher/_development_job_dispatcher.py

+ 10 - 1
taipy/core/_orchestrator/_dispatcher/_development_job_dispatcher.py

@@ -12,6 +12,8 @@
 import datetime
 from typing import Optional
 
+from ...common._check_dependencies import _TAIPY_ENTERPRISE_CORE_MODULE, _using_enterprise
+from ...common._utils import _load_fct
 from ...job.job import Job
 from .._abstract_orchestrator import _AbstractOrchestrator
 from ._job_dispatcher import _JobDispatcher
@@ -46,6 +48,13 @@ class _DevelopmentJobDispatcher(_JobDispatcher):
             job (Job^): The job to submit on an executor with an available worker.
         """
         job.execution_started_at = datetime.datetime.now()
-        rs = _TaskFunctionWrapper(job.id, job.task).execute()
+        if _using_enterprise():
+            task_fct_wrapper = _load_fct(
+                _TAIPY_ENTERPRISE_CORE_MODULE + "._orchestrator._dispatcher._task_function_wrapper",
+                "_TaskFunctionWrapper",
+            )
+        else:
+            task_fct_wrapper = _TaskFunctionWrapper
+        rs = task_fct_wrapper(job.id, job.task).execute()
         self._update_job_status(job, rs)
         job.execution_ended_at = datetime.datetime.now()