Merge pull request #511 from adrianeboyd/bugfix/torch-import-tensor

Update for torch 1.9.0
explosion · Jun 16, 2021 · 501552c · 501552c
2 parents 8881755 + 81b10a8
commit 501552c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/azure-pipelines.yml b/azure-pipelines.yml
@@ -91,7 +91,7 @@ jobs:
       pip install -r requirements.txt
       pip install tensorflow
       pip install "mxnet; sys_platform != 'win32'"
-      pip install "torch==1.8.1+cpu" -f https://download.pytorch.org/whl/torch_stable.html
+      pip install "torch==1.9.0+cpu" -f https://download.pytorch.org/whl/torch_stable.html
       pip install ipykernel pydot graphviz
       python -m ipykernel install --name thinc-notebook-tests --user
       python -m pytest --pyargs thinc --cov=thinc --cov-report=term

diff --git a/thinc/about.py b/thinc/about.py
@@ -1,2 +1,2 @@
-__version__ = "8.0.4"
+__version__ = "8.0.5"
 __release__ = True
diff --git a/thinc/util.py b/thinc/util.py
@@ -25,7 +25,7 @@
 
 try:  # pragma: no cover
     import torch
-    import torch.tensor
+    from torch import tensor
     import torch.utils.dlpack
 
     has_torch = True