[NemotronH] Fix weight-loading unit test broken by Puzzle support (#26522)

Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
Baizhou Zhang
2026-05-27 17:44:45 -07:00
committed by GitHub
co-authored by Claude Opus 4.7
parent 0abe6a85a5
commit 68e5b4fdd6
@@ -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 = {}