pyro-ppl / pyro / 5105007466

92%

push

github

make ProvenanceTensor behave more like a Tensor (closes #3218) (#3220)

* make ProvenanceTensor behave more like a Tensor (closes #3218)

the data is now stored in the Tensor itself instead of an attribute.
This fixes torch.to_tensor returning empty tensors when called with a
ProvenanceTensor and and a device as arguments

* fix compatibility with PyTorch 1.11

* make detach_provenance always return the exact same object

this is important when using Tensors as keys in a dict, e.g.
the Pyro param store

* preserve .unconstrained attribute in detach_provenance

* simplify

* add unit test

* simplify further

* use Tensor.as_subclass instead of modifying __class__

also remove unnecessary check in __init__

3 of 3 new or added lines in 1 file covered. (100.0%)

22685 of 24707 relevant lines covered (91.82%)

2.27 hits per line