diff --git a/python/oneflow/test/modules/test_sum.py b/python/oneflow/test/modules/test_sum.py index dd3a9f44d16..c9c4438f377 100644 --- a/python/oneflow/test/modules/test_sum.py +++ b/python/oneflow/test/modules/test_sum.py @@ -93,7 +93,7 @@ def test_sum_dtype(test_case): x, dim=np.random.randint(0, 3), keepdim=random_bool(), - dtype=random_dtype(["arithmetic", "complex"]), + dtype=random_dtype(["arithmetic"]), ) return y