Merge branch 'support_input_check' of https://github.com/Oneflow-Inc/…

…oneflow into support_input_check
Oneflow-Inc · May 29, 2024 · 0a78cdc · 0a78cdc
2 parents 656879e + 170a197
commit 0a78cdc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/oneflow/nn/modules/constant.py b/python/oneflow/nn/modules/constant.py
@@ -44,8 +44,8 @@ def __init__(
             self.device = flow.device(self.device)
         self.requires_grad = requires_grad
         size = _single(size)
-        assert (
-            all(s >= 0 for s in size)
+        assert all(
+            s >= 0 for s in size
         ), f"Trying to create tensor with negative dimension: {size}"
         if dtype is None:
             dtype = flow.get_default_dtype()