init

pytorch · Feb 4, 2025 · e49ea50 · e49ea50
1 parent 5630fc8
commit e49ea50
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/.github/workflows/wheels-windows.yml b/.github/workflows/wheels-windows.yml
@@ -43,7 +43,7 @@ jobs:
       - name: Upload wheel for download
         uses: actions/upload-artifact@v4
         with:
-          name: tensordict-batch.whl
+          name: tensordict-win-${{ matrix.python_version[0] }}.whl
           path: dist/*.whl
 
   test-wheel-windows:

diff --git a/tensordict/nn/distributions/composite.py b/tensordict/nn/distributions/composite.py
@@ -304,9 +304,7 @@ def maybe_deterministic_sample(dist):
             if hasattr(dist, "deterministic_sample"):
                 return dist.deterministic_sample
             else:
-                from tensordict.nn.probabilistic import (
-                    DETERMINISTIC_REGISTER,
-                )
+                from tensordict.nn.probabilistic import DETERMINISTIC_REGISTER
 
                 # Fallbacks
                 tdist = type(dist)

diff --git a/test/test_compile.py b/test/test_compile.py
@@ -6,6 +6,7 @@
 import contextlib
 import importlib.util
 import inspect
+import platform
 from pathlib import Path
 from typing import Any, Callable
 
@@ -43,6 +44,8 @@
 
 _v2_5 = TORCH_VERSION >= version.parse("2.5.0")
 
+_IS_OSX = platform.system() == "Darwin"
+
 
 def test_vmap_compile():
     # Since we monkey patch vmap we need to make sure compile is happy with it
@@ -952,6 +955,10 @@ def to_numpy(tensor):
 @pytest.mark.skipif(
     TORCH_VERSION <= version.parse("2.4.1"), reason="requires torch>=2.5"
 )
+@pytest.mark.skipif(
+    (TORCH_VERSION <= version.parse("2.7.0")) and _IS_OSX,
+    reason="requires torch>=2.7 ons OSX",
+)
 @pytest.mark.parametrize("compiled", [False, True])
 class TestCudaGraphs:
     @pytest.fixture(scope="class", autouse=True)