test(speechlm2): use consistent SALM imports

Signed-off-by: Piotr Żelasko <pzelasko@nvidia.com>

Author: pzelasko

tests/collections/speechlm2/test_vllm_plugin.py

@@ -1312,7 +1312,7 @@ class TestMTPPlugin:
     @pytest.fixture(autouse=True)
     def restore_original_override(self):
         """Keep the process-local fallback hook isolated between tests."""
-        import nemo.collections.speechlm2.vllm.salm as salm_module
+        from nemo.collections.speechlm2.vllm import salm as salm_module
 
         original_override = salm_module._ORIGINAL_VLLM_HF_CONFIG_OVERRIDE
         yield
@@ -1420,7 +1420,7 @@ def test_patched_override_is_pickleable_for_spawned_engine(self, monkeypatch):
         from transformers import AutoConfig
         from vllm.config.speculative import SpeculativeConfig
 
-        import nemo.collections.speechlm2.vllm.salm as salm_module
+        from nemo.collections.speechlm2.vllm import salm as salm_module
         from nemo.collections.speechlm2.vllm.salm import register
 
         monkeypatch.setattr(
@@ -1456,7 +1456,7 @@ def test_patched_override_lazily_captures_native_hook_in_spawn_child(self, monke
         """A fresh spawn import should delegate unrelated configs to vLLM's native hook."""
         from vllm.config.speculative import SpeculativeConfig
 
-        import nemo.collections.speechlm2.vllm.salm as salm_module
+        from nemo.collections.speechlm2.vllm import salm as salm_module
 
         original_calls = []
 
@@ -1482,7 +1482,7 @@ def test_patched_override_rejects_missing_native_hook(self, monkeypatch):
         """A corrupted install must fail clearly instead of recursing into our override."""
         from vllm.config.speculative import SpeculativeConfig
 
-        import nemo.collections.speechlm2.vllm.salm as salm_module
+        from nemo.collections.speechlm2.vllm import salm as salm_module
 
         monkeypatch.setattr(salm_module, "_ORIGINAL_VLLM_HF_CONFIG_OVERRIDE", None)
         monkeypatch.setattr(
@@ -1560,7 +1560,7 @@ def _recording_orig(cfg):
             original_calls.append(cfg)
             return cfg
 
-        import nemo.collections.speechlm2.vllm.salm as salm_module
+        from nemo.collections.speechlm2.vllm import salm as salm_module
 
         monkeypatch.setattr(salm_module, "_ORIGINAL_VLLM_HF_CONFIG_OVERRIDE", None)
         monkeypatch.setattr(SpeculativeConfig, "hf_config_override", staticmethod(_recording_orig))
@@ -1579,7 +1579,7 @@ def test_patched_override_depth_without_enabled_flag_falls_through(self, monkeyp
         from transformers import AutoConfig
         from vllm.config.speculative import SpeculativeConfig
 
-        import nemo.collections.speechlm2.vllm.salm as salm_module
+        from nemo.collections.speechlm2.vllm import salm as salm_module
         from nemo.collections.speechlm2.vllm.salm import register
 
         monkeypatch.setattr(
@@ -1611,7 +1611,7 @@ def test_patched_override_explicitly_disabled_mtp_falls_through(self, monkeypatc
         from transformers import AutoConfig
         from vllm.config.speculative import SpeculativeConfig
 
-        import nemo.collections.speechlm2.vllm.salm as salm_module
+        from nemo.collections.speechlm2.vllm import salm as salm_module
         from nemo.collections.speechlm2.vllm.salm import register
 
         monkeypatch.setattr(
@@ -1690,7 +1690,7 @@ def test_mtp_override_reregistration_preserves_first_native_hook(self, monkeypat
         from transformers import AutoConfig
         from vllm.config.speculative import SpeculativeConfig
 
-        import nemo.collections.speechlm2.vllm.salm as salm_module
+        from nemo.collections.speechlm2.vllm import salm as salm_module
         from nemo.collections.speechlm2.vllm.salm import register
 
         monkeypatch.setattr(