fix test

Signed-off-by: Jennifer Chen <jennifchen@nvidia.com>

Author: jenchen13

tests/unit/torch/export/test_hf_checkpoint_utils.py

@@ -90,21 +90,21 @@ def test_copy_hf_ckpt_remote_code_hub_id_offline_uses_cache(tmp_path, monkeypatc
     dst_dir = tmp_path / "dst"
     snapshot_dir = tmp_path / "snapshot"
     snapshot_dir.mkdir()
-    (snapshot_dir / "ultra_v3_reasoning_parser.py").write_text("# parser")
+    (snapshot_dir / "nemotron_reasoning_parser.py").write_text("# parser")
 
     monkeypatch.setenv("HF_HUB_OFFLINE", "1")
     with patch(
         "modelopt.torch.export.plugins.hf_checkpoint_utils.snapshot_download",
         return_value=str(snapshot_dir),
     ) as mock_sd:
-        copy_hf_ckpt_remote_code("nvidia/nemotron-ultra-rl-041326", dst_dir)
+        copy_hf_ckpt_remote_code("nvidia/NVIDIA-Nemotron-3-Nano-30B-A3B-BF16", dst_dir)
 
     mock_sd.assert_called_once_with(
-        repo_id="nvidia/nemotron-ultra-rl-041326",
+        repo_id="nvidia/NVIDIA-Nemotron-3-Nano-30B-A3B-BF16",
         allow_patterns=["*.py"],
         local_files_only=True,
     )
-    assert (dst_dir / "ultra_v3_reasoning_parser.py").read_text() == "# parser"
+    assert (dst_dir / "nemotron_reasoning_parser.py").read_text() == "# parser"
 
 
 def test_copy_hf_ckpt_remote_code_hub_id_offline_missing_cache_raises(tmp_path, monkeypatch):
@@ -115,4 +115,6 @@ def test_copy_hf_ckpt_remote_code_hub_id_offline_missing_cache_raises(tmp_path,
         side_effect=LocalEntryNotFoundError("missing"),
     ):
         with pytest.raises(RuntimeError, match="HF_HUB_OFFLINE"):
-            copy_hf_ckpt_remote_code("nvidia/nemotron-ultra-rl-041326", tmp_path / "dst")
+            copy_hf_ckpt_remote_code(
+                "nvidia/NVIDIA-Nemotron-3-Nano-30B-A3B-BF16", tmp_path / "dst"
+            )