docstring changes

sandialabs · May 7, 2024 · 1ec6909 · 1ec6909
1 parent 5a1be5d
commit 1ec6909
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pygsti/modelmembers/torchable.py b/pygsti/modelmembers/torchable.py
@@ -28,10 +28,10 @@ def torch_base(sd : Tuple, t_param : Tensor) -> Tensor:
         """
         Suppose "obj" is an instance of some Torchable subclass. If we compute
 
-            sd = obj.stateless_data()
             vec = obj.to_vector()
             t_param = torch.from_numpy(vec)
-            t = type(obj).torch_base(sd, t_param, grad)
+            sd = obj.stateless_data()
+            t = type(obj).torch_base(sd, t_param)
 
         then t will be a PyTorch Tensor that represents "obj" in a canonical numerical way.