From 62c70b31639bae91d92ccab184b38c8b8391d009 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Sun, 14 Oct 2018 23:03:06 +0200 Subject: [PATCH] Import prefer_gpu and require_gpu functions from Thinc --- spacy/__init__.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/spacy/__init__.py b/spacy/__init__.py index b8cf63b1c..411883eae 100644 --- a/spacy/__init__.py +++ b/spacy/__init__.py @@ -4,6 +4,9 @@ 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, require_gpu + from .cli.info import info as cli_info from .glossary import explain from .about import __version__