cloudpickle?

DecBayComp · Mar 30, 2021 · 37d3b23 · 37d3b23
1 parent e4aacd2
commit 37d3b23
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -55,7 +55,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        python -m pip install pytest dill
+        python -m pip install pytest dill cloudpickle
         python -m pip install .[roi]
     - name: Test with pytest (Windows)
       if: ${{ runner.os == 'Windows' }}

diff --git a/tramway/core/parallel/__init__.py b/tramway/core/parallel/__init__.py
@@ -16,6 +16,10 @@
     import dill
 except ImportError:
     pass
+try:
+    import cloudpickle
+except ImportError:
+    pass
 import multiprocessing
 try:
     import queue

diff --git a/tramway/inference/base.py b/tramway/inference/base.py
@@ -28,6 +28,10 @@
     import dill
 except ImportError:
     pass
+try:
+    import cloudpickle
+except ImportError:
+    pass
 from multiprocessing import Pool, Lock
 
 import os # for os.name