add unit test in support of complex64 data type

huggingface · Nov 18, 2023 · e927b83 · e927b83
1 parent 0d940f8
commit e927b83
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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.