Skip to content

Commit

Permalink
force multiprocessing instead of loky
Browse files Browse the repository at this point in the history
Signed-off-by: Lukas Heumos <[email protected]>
  • Loading branch information
Zethson committed Feb 21, 2025
1 parent 633771f commit 07b8fc2
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
4 changes: 4 additions & 0 deletions pertpy/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,13 @@

import warnings

import joblib
from matplotlib import MatplotlibDeprecationWarning
from numba import NumbaDeprecationWarning

# Loky seems to be broken at the moment
joblib.parallel.DEFAULT_BACKEND = "multiprocessing"

warnings.filterwarnings("ignore", category=NumbaDeprecationWarning)
warnings.filterwarnings("ignore", category=MatplotlibDeprecationWarning)
warnings.filterwarnings("ignore", category=SyntaxWarning)
Expand Down
6 changes: 1 addition & 5 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -63,12 +63,8 @@ dependencies = [
"pubchempy",
"pyarrow",
"blitzgsea",
"scikit-learn>=1.4", # https://github.com/scverse/pertpy/issues/700
"lamin_utils",
"joblib",
"scipy==1.15.0",
"torch==2.5.1",
"toytree==3.0.6",
"triton==3.1.0"
]

[project.optional-dependencies]
Expand Down

0 comments on commit 07b8fc2

Please sign in to comment.