Skip to content

GitLab

Unverified Commit a33d6e2d authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

[FBcode->GH] [JIT] Add reference semantics to TorchScript classes (#44324) (#4166) 


Co-authored-by: default avatarMeghan Lele <meghanl@fb.com>
parent 23b87603
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
test/test_transforms_tensor.py
View file @ a33d6e2d
...@@ -649,7 +649,7 @@ def test_compose(device): ...@@ -649,7 +649,7 @@ def test_compose(device):
t = T.Compose([ t = T.Compose([
lambda x: x, lambda x: x,
]) ])
with pytest.raises(RuntimeError, match="Could not get name of python class object"): with pytest.raises(RuntimeError, match="cannot call a value of type 'Tensor'"):
torch.jit.script(t) torch.jit.script(t)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment