Removing () from assertion. (#307)

The parenthesis alter the evaluation of the assert() and it will always evaluate to True.

Removing () from assertion. (#307)
The parenthesis alter the evaluation of the assert() and it will always evaluate to True.
c35e9441 · Shaden Smith · GitHub · 9d07d756 · c35e9441
Unverified Commit c35e9441 authored Aug 07, 2020 by Shaden Smith Committed by GitHub Aug 07, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

deepspeed/pt/zero_optimizer_stage1.py deepspeed/pt/zero_optimizer_stage1.py +1 -2

No files found.
--- a/deepspeed/pt/zero_optimizer_stage1.py
+++ b/deepspeed/pt/zero_optimizer_stage1.py
@@ -40,8 +40,7 @@ def flatten_dense_tensors_sub_partition_aligned(tensor_list,
                                                dp,
                                                max_elements_per_comm,
                                                pg):
-    assert (max_elements_per_comm >= dp,
-            f"max_elements_per_comm {max_elements_per_comm} < dp {dp}")
+    assert max_elements_per_comm >= dp, f"max_elements_per_comm {max_elements_per_comm} < dp {dp}"

    num_elements = sum(t.numel() for t in tensor_list)
    log_dist("Total number of elements in model: {}, max elements per com: {}".format(