fixed up test_pickle

Author: tclose

pydra/workers/cf.py

@@ -46,7 +46,9 @@ class Worker(base.Worker):
     """A worker to execute in parallel using Python's concurrent futures."""
 
     n_procs: int = attrs.field(factory=get_available_cpus)
-    pool: cf.ProcessPoolExecutor = attrs.field()
+    pool: cf.ProcessPoolExecutor = attrs.field(
+        eq=False, init=False, hash=False, repr=False
+    )
 
     @pool.default
     def _pool_default(self) -> cf.ProcessPoolExecutor:

pydra/workers/tests/test_pickle.py

@@ -1,13 +1,43 @@
 import pickle as pkl
 
-from pydra.workers import debug, cf, sge, slurm, dask, base
+from pydra.workers import debug, cf, sge, slurm
 
 
 def test_pickle_debug_worker():
     """
-    Test pickling of DebugWorker
+    Test pickling of debug.Worker
     """
     worker = debug.Worker()
     worker2 = pkl.loads(pkl.dumps(worker))
     assert worker2.loop is None
     assert worker == worker2
+
+
+def test_pickle_cf_worker():
+    """
+    Test pickling of cf.Worker
+    """
+    worker = cf.Worker()
+    worker2 = pkl.loads(pkl.dumps(worker))
+    assert worker2.loop is None
+    assert worker == worker2
+
+
+def test_pickle_sge_worker():
+    """
+    Test pickling of sge.Worker
+    """
+    worker = sge.Worker()
+    worker2 = pkl.loads(pkl.dumps(worker))
+    assert worker2.loop is None
+    assert worker == worker2
+
+
+def test_pickle_slurm_worker():
+    """
+    Test pickling of DebugWorker
+    """
+    worker = slurm.Worker()
+    worker2 = pkl.loads(pkl.dumps(worker))
+    assert worker2.loop is None
+    assert worker == worker2