diff --git a/bindings/python/tests/test_simple.py b/bindings/python/tests/test_simple.py index c46320c0..ca70684f 100644 --- a/bindings/python/tests/test_simple.py +++ b/bindings/python/tests/test_simple.py @@ -215,6 +215,7 @@ def test_torch_example(self): tensors = { "a": torch.zeros((2, 2)), "b": torch.zeros((2, 3), dtype=torch.uint8), + "c": torch.randn(2,2, dtype=torch.complex64), } # Saving modifies the tensors to type numpy, so we must copy for the # test to be correct.