Fix provenance tensor to work with PyTorch 1.11 (#588)

pyro-ppl · Mar 17, 2022 · 821c1b4 · 821c1b4
1 parent 7ea6ac5
commit 821c1b4
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/funsor/torch/provenance.py b/funsor/torch/provenance.py
@@ -14,9 +14,7 @@ class ProvenanceTensor(torch.Tensor):
     def __new__(cls, data, provenance=frozenset(), **kwargs):
         if not provenance:
             return data
-        instance = torch.Tensor.__new__(cls)
-        instance.__init__(data, provenance)
-        return instance
+        return torch.Tensor.__new__(cls)
 
     def __init__(self, data, provenance=frozenset()):
         assert isinstance(provenance, frozenset)