From 68e5b4fdd66bd9cca41156d6e2aa75a857d58ff6 Mon Sep 17 00:00:00 2001 From: Baizhou Zhang Date: Wed, 27 May 2026 17:44:45 -0700 Subject: [PATCH] [NemotronH] Fix weight-loading unit test broken by Puzzle support (#26522) Co-authored-by: Claude Opus 4.7 (1M context) --- test/registered/unit/models/test_nemotron_h_weight_loading.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/registered/unit/models/test_nemotron_h_weight_loading.py b/test/registered/unit/models/test_nemotron_h_weight_loading.py index 0f886cdda..6a1221ff3 100644 --- a/test/registered/unit/models/test_nemotron_h_weight_loading.py +++ b/test/registered/unit/models/test_nemotron_h_weight_loading.py @@ -35,7 +35,7 @@ class _FakeParam: class TestNemotronHWeightLoading(unittest.TestCase): def _make_minimal_model(self, named_parameters=()): model = object.__new__(NemotronHForCausalLM) - model.config = SimpleNamespace(n_routed_experts=2) + model.config = SimpleNamespace(n_routed_experts=2, max_n_routed_experts=2) model.model = SimpleNamespace() model.pp_group = _FakePPGroup() model.remap_prefix = {}