mirror of
https://github.com/explosion/spaCy.git
synced 2024-12-28 19:06:33 +03:00
d94ddd5686
* Auto-detect package dependencies in spacy package * Add simple get_third_party_dependencies test * Import packages_distributions explicitly * Inline packages_distributions * Fix docstring [ci skip] * Relax catalogue requirement * Move importlib_metadata to spacy.compat with note * Include license information [ci skip]
49 lines
1.2 KiB
Python
49 lines
1.2 KiB
Python
"""Helpers for Python and platform compatibility."""
|
|
import sys
|
|
from thinc.util import copy_array
|
|
|
|
try:
|
|
import cPickle as pickle
|
|
except ImportError:
|
|
import pickle
|
|
|
|
try:
|
|
import copy_reg
|
|
except ImportError:
|
|
import copyreg as copy_reg
|
|
|
|
try:
|
|
from cupy.cuda.stream import Stream as CudaStream
|
|
except ImportError:
|
|
CudaStream = None
|
|
|
|
try:
|
|
import cupy
|
|
except ImportError:
|
|
cupy = None
|
|
|
|
try: # Python 3.8+
|
|
from typing import Literal
|
|
except ImportError:
|
|
from typing_extensions import Literal # noqa: F401
|
|
|
|
# Important note: The importlib_metadata "backport" includes functionality
|
|
# that's not part of the built-in importlib.metadata. We should treat this
|
|
# import like the built-in and only use what's available there.
|
|
try: # Python 3.8+
|
|
import importlib.metadata as importlib_metadata
|
|
except ImportError:
|
|
from catalogue import _importlib_metadata as importlib_metadata # noqa: F401
|
|
|
|
from thinc.api import Optimizer # noqa: F401
|
|
|
|
pickle = pickle
|
|
copy_reg = copy_reg
|
|
CudaStream = CudaStream
|
|
cupy = cupy
|
|
copy_array = copy_array
|
|
|
|
is_windows = sys.platform.startswith("win")
|
|
is_linux = sys.platform.startswith("linux")
|
|
is_osx = sys.platform == "darwin"
|