spaCy/spacy/__init__.py

# coding: utf8
from __future__ import unicode_literals
import warnings
warnings.filterwarnings("ignore", message="numpy.dtype size changed")
warnings.filterwarnings("ignore", message="numpy.ufunc size changed")

# These are imported as part of the API
from thinc.neural.util import prefer_gpu

from .cli.info import info as cli_info
from .glossary import explain
from .about import __version__
from .errors import Warnings, deprecation_warning
from . import util


def load(name, **overrides):
    depr_path = overrides.get('path')
    if depr_path not in (True, False, None):
        deprecation_warning(Warnings.W001.format(path=depr_path))
    return util.load_model(name, **overrides)


def blank(name, **kwargs):
    LangClass = util.get_lang_class(name)
    return LangClass(**kwargs)


def info(model=None, markdown=False, silent=False):
    return cli_info(model, markdown, silent)

def require_gpu():
    from thinc.v2v import Model
    from thinc.neural.ops import CupyOps
    if CupyOps.xp is None:
        raise ValueError(
            "GPU is not accessible. Check your LD_LIBRARY_PATH enironment variable "
            "and check that thinc was installed with GPU, e.g. thinc[cuda]")
    Model.Ops = CupyOps
    Model.ops = CupyOps()
    return True