test fixes

Signed-off-by: Patryk Saffer <[email protected]>

Author: Patryk999

tests/v1/e2e/test_eplb_offline.py

@@ -48,7 +48,7 @@ def test_eplb_model(
         )
 
         # Save EPLB statistics to disk
-        eplb_config_save = EPLBConfig(save_dir="/tmp")
+        eplb_config_save = EPLBConfig(save_load_window=True, save_dir="/tmp")
         llm = LLM(eplb_config=eplb_config_save, **llm_args)
         llm.generate(test_prompts, sampling_config)
         del llm
@@ -57,6 +57,7 @@ def test_eplb_model(
 
         # Load EPLB statistics from disk
         eplb_config_load = EPLBConfig(
+            load_initial_load_window=True,
             load_path="/tmp/global_expert_load_window_i0.safetensors"
         )
         llm = LLM(eplb_config=eplb_config_load, **llm_args)

vllm/model_executor/layers/fused_moe/unquantized_fused_moe_method.py

@@ -283,6 +283,7 @@ def apply(
         apply_router_weight_on_input: bool = False,
         activation: str = "silu",
         enable_eplb: bool = False,
+        eplb_static: bool = False,
         expert_load_view: torch.Tensor | None = None,
         logical_to_physical_map: torch.Tensor | None = None,
         logical_replica_count: torch.Tensor | None = None,