Skip torch test on Python 3.13 / Windows

pola-rs · Jan 5, 2025 · d07df04 · d07df04
1 parent c3a4227
commit d07df04
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/py-polars/requirements-ci.txt b/py-polars/requirements-ci.txt
@@ -3,6 +3,6 @@
 # (installable via `make requirements-all`)
 # -------------------------------------------------------
 --extra-index-url https://download.pytorch.org/whl/cpu
-torch
+torch; python_version < '3.13' or platform_system != 'Windows'  # torch provides no wheel for Python 3.13 on Windows
 jax[cpu]
 pyiceberg>=0.5.0
diff --git a/py-polars/tests/unit/ml/test_to_torch.py b/py-polars/tests/unit/ml/test_to_torch.py
@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import sys
 from typing import Any
 
 import pytest
@@ -12,7 +13,13 @@
 # ensures the tests aren't run locally; this avoids premature local import)
 torch, _ = _lazy_import("torch")
 
-pytestmark = pytest.mark.ci_only
+pytestmark = [
+    pytest.mark.ci_only,
+    pytest.mark.skipif(
+        sys.platform == "win32" and sys.version_info >= (3, 13),
+        reason="Torch does not ship wheels for Python 3.13 on Windows",
+    ),
+]
 
 
 @pytest.fixture