From 5cbc9f4573686857cd5b2cfb8cc38fb51ff2a5a2 Mon Sep 17 00:00:00 2001 From: Paul O'Leary McCann Date: Tue, 24 May 2022 16:02:39 +0900 Subject: [PATCH] Use thinc.util.has_torch --- spacy/ml/models/__init__.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/spacy/ml/models/__init__.py b/spacy/ml/models/__init__.py index b01721964..4368a556d 100644 --- a/spacy/ml/models/__init__.py +++ b/spacy/ml/models/__init__.py @@ -7,10 +7,8 @@ from .textcat import * # noqa from .tok2vec import * # noqa # some models require Torch -try: - import torch +from thinc.util import has_torch +if has_torch: from .coref import * #noqa from .span_predictor import * #noqa -except ImportError: - pass