[Bug] Fix parallel mode tests

xrsrke · Dec 10, 2023 · b74a92d · b74a92d
1 parent dfda88c
commit b74a92d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/distributed/test_parallel_mode.py b/tests/distributed/test_parallel_mode.py
@@ -6,7 +6,7 @@ def test_parallel_mode():
     assert hasattr(ParallelMode, "TENSOR")
     assert hasattr(ParallelMode, "PIPELINE")
     assert hasattr(ParallelMode, "DATA")
-    assert hasattr(ParallelMode, "EXPERT")
+    assert hasattr(ParallelMode, "EXPERT_DATA")
 
     assert ParallelMode.GLOBAL == ParallelMode.GLOBAL
     assert ParallelMode.GLOBAL != ParallelMode.TENSOR