@@ -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