TPU simple tests

Signed-off-by: Vladimir Suvorov <suvorovv@google.com>

Author: mydatascience

.github/workflows/tpu-tests.yml

@@ -72,8 +72,6 @@ jobs:
         # Config tests that passed
         python -m pytest tests/sft/config_test.py -v --tb=short
         
-        # DPO tests that passed
-        python -m pytest tests/sft/dpo/dpo_trainer_test.py::DpoTrainerTest::test_dpo_trainer_chosen_reject_equal_length -v --tb=short
         
         # Checkpoint manager test that passed
         python -m pytest tests/sft/checkpoint_manager_test.py::CheckpointManagerTest::test_empty_root_directory -v --tb=short