Code cleanup (#234)

This commit is contained in:
Matthew Tancik authored and GitHub committed 2022-07-31 18:22:57 -07:00
1 parent 8b414bafab
commit e35781fe85
14 files changed
+46 -68

No files matched your search

+1 -1
View File
@@ -48,7 +48,7 @@ def test_broadcasting():
assert tensor_dataclass.b.shape == (4, 6, 2)
a = torch.ones((4, 6, 3))
b = torch.ones((2))
b = torch.ones(2)
tensor_dataclass = TestTensorDataclass(a=a, b=b)
assert tensor_dataclass.b.shape == (4, 6, 2)