Rename PyTorch to Torch

At some point PyTorch was renamed Torch in the pip repos, this changes how the package is pulled mostly for RTD, since most users will be pulling the package through Anaconda where the package is still named PyTorch
JeffersonLab · Jul 13, 2022 · c04a43c · c04a43c
1 parent 5944081
commit c04a43c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/.readthedocs.yml b/.readthedocs.yml
@@ -15,5 +15,5 @@ python:
     - method: pip
       path: .
       extra_requirements:
-        - pytorch
+        - torch
         - emcee
diff --git a/PyPWA/libs/fit/likelihoods.py b/PyPWA/libs/fit/likelihoods.py
@@ -34,6 +34,7 @@
     TORCH_AVAIL = True
 except ImportError:
     torch = npy
+    torch.Tensor = npy.ndarray
     TORCH_AVAIL = False
 
 from PyPWA import info as _info

diff --git a/setup.py b/setup.py
@@ -45,7 +45,7 @@
 
 extras = {
     "emcee": ["emcee"],     # Provides MCMC fitting
-    "pytorch": ["pytorch"]  # Provides PyTorch support
+    "torch": ["torch"]  # Provides PyTorch support
 }
 
 tests = [
-Original file line number
+Diff line change
@@ Expand Up / @@ -15,5 +15,5 @@ python: @@
         - method: pip
           path: .
           extra_requirements:
-            - pytorch
+            - torch
             - emcee