From c9233a5a1f34b2e8397d40abfde283412bc79b6e Mon Sep 17 00:00:00 2001 From: Paul O'Leary McCann Date: Tue, 24 May 2022 17:28:27 +0900 Subject: [PATCH] Import torch from thinc --- spacy/ml/models/coref.py | 3 +-- spacy/ml/models/span_predictor.py | 3 +-- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/spacy/ml/models/coref.py b/spacy/ml/models/coref.py index b4d8030e8..ca9011577 100644 --- a/spacy/ml/models/coref.py +++ b/spacy/ml/models/coref.py @@ -1,10 +1,9 @@ from typing import List, Tuple -import torch from thinc.api import Model, chain from thinc.api import PyTorchWrapper, ArgsKwargs from thinc.types import Floats2d, Ints2d, Ints1d -from thinc.util import xp2torch, torch2xp +from thinc.util import torch, xp2torch, torch2xp from ...tokens import Doc from ...util import registry diff --git a/spacy/ml/models/span_predictor.py b/spacy/ml/models/span_predictor.py index 7375c2153..1ded9c3c7 100644 --- a/spacy/ml/models/span_predictor.py +++ b/spacy/ml/models/span_predictor.py @@ -1,10 +1,9 @@ from typing import List, Tuple -import torch from thinc.api import Model, chain, tuplify from thinc.api import PyTorchWrapper, ArgsKwargs from thinc.types import Floats2d, Ints1d -from thinc.util import xp2torch, torch2xp +from thinc.util import torch, xp2torch, torch2xp from ...tokens import Doc from ...util import registry