Update on "[MPS] Fix bug when value is of complex"

qqaatw · qqaatw · commit a3a5c61cee28 · 2023-10-24T12:57:21.000-07:00
[ghstack-poisoned]
diff --git a/test/test_mps.py b/test/test_mps.py
@@ -1403,7 +1403,7 @@ def helper(val, shape, dtype):
 
         helper(0, [1024], torch.float32)
         helper(0.2, [2, 3], torch.float32)
-        helper(0.2+0.5j, [2, 3], torch.complex64)
+        helper(0.2 + 0.5j, [2, 3], torch.complex64)
 
     def test_fill_storage_offset(self):
         shape = [2, 10]
