From 9c0fa6022047cb88c75bdb534d8c42a24d85642a Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 11 Aug 2025 16:26:42 -0400 Subject: [PATCH] fsdp2 w evals fixed upstream --- tests/monkeypatch/test_trainer_loss_calc.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/tests/monkeypatch/test_trainer_loss_calc.py b/tests/monkeypatch/test_trainer_loss_calc.py index de3e92621..c72cb621b 100644 --- a/tests/monkeypatch/test_trainer_loss_calc.py +++ b/tests/monkeypatch/test_trainer_loss_calc.py @@ -3,7 +3,6 @@ import unittest from axolotl.monkeypatch.transformers.trainer_loss_calc import ( - check_evaluation_loop_is_fsdp2_patchable, check_evaluation_loop_is_patchable, check_maybe_log_save_evaluate_is_patchable, ) @@ -20,7 +19,6 @@ class TestTrainerLossCalc(unittest.TestCase): the patched code changes upstream. """ assert check_evaluation_loop_is_patchable() - assert check_evaluation_loop_is_fsdp2_patchable() assert check_maybe_log_save_evaluate_is_patchable()