use single variable for returned value

the old impl raised error "too many values to unpack (expected 1)"

use single variable for returned value
the old impl raised error "too many values to unpack (expected 1)"
318513ae · Jiezhong Qiu · GitHub · 28bfe689 · 318513ae
Unverified Commit 318513ae authored Jun 17, 2021 by Jiezhong Qiu Committed by GitHub Jun 17, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

fmoe/gates/utils.py fmoe/gates/utils.py +1 -1

No files found.
--- a/fmoe/gates/utils.py
+++ b/fmoe/gates/utils.py
@@ -16,7 +16,7 @@ def limit_by_capacity(topk_idx, num_expert, world_size, capacity):
        new_gec = fmoe_native.limit_by_capacity(gec, capacity,
                num_expert, world_size)
        if world_size > 1:
-            new_lec, = fmoe_native.expert_exchange(new_gec, num_expert,
+            new_lec = fmoe_native.expert_exchange(new_gec, num_expert,
                    world_size)
        else:
            new_lec = new_gec