fix: update broken import path in medusa unit test (#515)

Author: rebel-jinhwan

tests/torch_compile/e2e/v1/spec_decode/utils.py

@@ -51,6 +51,7 @@ def _load_json(path: str | Path) -> dict[str, Any]:
     with Path(path).open("r", encoding="utf-8") as f:
         return json.load(f)
 
+
 def _is_vllm_medusa_config(config_dict: dict[str, Any]) -> bool:
     return config_dict.get("model_type") == "medusa" and isinstance(
         config_dict.get("num_heads"), int

tests/torch_compile/unit/v1/spec_decode/test_eagle.py

@@ -149,8 +149,8 @@ def _set_positions(num_tokens: int, positions: torch.Tensor) -> None:
         fake.positions[:num_tokens] = positions
 
     fake._set_positions = _set_positions
-    fake.set_inputs_first_pass = (
-        lambda **kwargs: RBLNEagleProposer.set_inputs_first_pass(fake, **kwargs)
+    fake.set_inputs_first_pass = lambda **kwargs: (
+        RBLNEagleProposer.set_inputs_first_pass(fake, **kwargs)
     )
     return fake, builder
 

tests/torch_compile/unit/v1/spec_decode/test_medusa.py

@@ -24,8 +24,8 @@
 import torch
 from vllm.v1.spec_decode.medusa import MedusaProposer
 
-import vllm_rbln.v1.spec_decoding.medusa as medusa_module
-from vllm_rbln.v1.spec_decoding.medusa import RBLNMedusaProposer
+import vllm_rbln.v1.spec_decode.medusa as medusa_module
+from vllm_rbln.v1.spec_decode.medusa import RBLNMedusaProposer
 
 
 class FakeMedusaModel: