Fixing missing function set_executor_gpu from import

Signed-off-by: Julio Faracco <[email protected]>
discovery-unicamp · Mar 30, 2024 · 7905cb8 · 7905cb8
1 parent 21baacb
commit 7905cb8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dasf/pipeline/executors/dask.py b/dasf/pipeline/executors/dask.py
@@ -27,6 +27,7 @@
     get_worker_info,
     is_dask_gpu_supported,
     is_gpu_supported,
+    set_executor_gpu,
     set_executor_multi_cpu,
     set_executor_multi_gpu,
 )