skip this one

Author: ArthurZucker

tests/models/ernie4_5_moe/test_modeling_ernie4_5_moe.py

@@ -90,6 +90,7 @@ def test_flash_attn_2_equivalence(self):
                 assert torch.allclose(logits_fa, logits, atol=1e-2, rtol=1e-2)
 
     # Ignore copy
+    @unittest.skip("TODO @ArthurZucker investigate later on")
     def test_load_balancing_loss(self):
         r"""
         Let's make sure we can actually compute the loss and do a backward on it.