Skip to content

Commit

Permalink
ignore test windows
Browse files Browse the repository at this point in the history
  • Loading branch information
qgallouedec committed Feb 4, 2025
1 parent 9d571bf commit eaf5370
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions tests/test_grpo_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.

import sys
import tempfile
import unittest

Expand Down Expand Up @@ -391,6 +392,7 @@ def test_training_vllm(self):
new_param = trainer.model.get_parameter(n)
self.assertFalse(torch.equal(param, new_param), f"Parameter {n} has not changed.")

@unittest.skipIf(sys.platform.startswith("win"), "Skipping on Windows") # compiling seems to be broken on Windows
def test_training_torch_compile(self):
dataset = load_dataset("trl-internal-testing/zen", "standard_prompt_only", split="train")

Expand Down

0 comments on commit eaf5370

Please sign in to comment.