Skip to content

Commit 7abd810

Browse files
committed
fix test
Signed-off-by: HuiyingLi <willwin.lee@gmail.com>
1 parent d34b943 commit 7abd810

File tree

1 file changed

+17
-5
lines changed

1 file changed

+17
-5
lines changed

tests/unit_tests/recipes/test_train_ft.py

Lines changed: 17 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -501,11 +501,23 @@ def _patch_setup_minimals(monkeypatch, patch_mock: MagicMock):
501501
lambda *a, **k: SimpleNamespace(log=lambda *a, **k: None, close=lambda: None),
502502
)
503503

504-
# No-op logging helpers
505-
monkeypatch.setattr("nemo_automodel.recipes.llm.train_ft._log_experiment_details", lambda self: None)
506-
monkeypatch.setattr("nemo_automodel.recipes.llm.train_ft._log_library_versions", lambda self: None)
507-
monkeypatch.setattr("nemo_automodel.recipes.llm.train_ft._log_model_and_optimizer_details", lambda *a, **k: None)
508-
monkeypatch.setattr("nemo_automodel.recipes.llm.train_ft._setup_qat", lambda *a, **k: (None, None, None))
504+
# No-op logging helpers on the recipe class
505+
monkeypatch.setattr(
506+
"nemo_automodel.recipes.llm.train_ft.TrainFinetuneRecipeForNextTokenPrediction._log_experiment_details",
507+
lambda self: None,
508+
)
509+
monkeypatch.setattr(
510+
"nemo_automodel.recipes.llm.train_ft.TrainFinetuneRecipeForNextTokenPrediction._log_library_versions",
511+
lambda self: None,
512+
)
513+
monkeypatch.setattr(
514+
"nemo_automodel.recipes.llm.train_ft.TrainFinetuneRecipeForNextTokenPrediction._log_model_and_optimizer_details",
515+
lambda *a, **k: None,
516+
)
517+
monkeypatch.setattr(
518+
"nemo_automodel.recipes.llm.train_ft.TrainFinetuneRecipeForNextTokenPrediction._setup_qat",
519+
lambda *a, **k: (None, None, None),
520+
)
509521
monkeypatch.setattr("nemo_automodel.recipes.llm.train_ft.TrainFinetuneRecipeForNextTokenPrediction.load_checkpoint", lambda *a, **k: None)
510522
monkeypatch.setattr("nemo_automodel.recipes.llm.train_ft.TrainFinetuneRecipeForNextTokenPrediction._log_step_scheduler_details", lambda *a, **k: None)
511523

0 commit comments

Comments
 (0)