Skip to content

Commit

Permalink
fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
lu-wang-dl committed May 24, 2024
1 parent 371f911 commit 92cf975
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion joblibspark/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,6 @@ def create_resource_profile(num_cpus_worker_node, num_gpus_worker_node):
from pyspark.resource.requests import TaskResourceRequests
except ImportError:
pass

task_res_req = TaskResourceRequests().cpus(num_cpus_worker_node)
if num_gpus_worker_node > 0:
task_res_req = task_res_req.resource("gpu", num_gpus_worker_node)
Expand Down

0 comments on commit 92cf975

Please sign in to comment.