fix test

joblib · May 23, 2024 · 7250ea9 · 7250ea9
1 parent c9cade9
commit 7250ea9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/joblibspark/backend.py b/joblibspark/backend.py
@@ -69,8 +69,8 @@ class SparkDistributedBackend(ParallelBackendBase, AutoBatchingMixin):
     """
 
     def __init__(self,
-                 num_cpus_worker_node: Optional[int],
-                 num_gpus_worker_node: Optional[int],
+                 num_cpus_worker_node: Optional[int] = None,
+                 num_gpus_worker_node: Optional[int] = None,
                  **backend_args):
         # pylint: disable=super-with-arguments
         super(SparkDistributedBackend, self).__init__(**backend_args)