Disabled pytorch for 3.12 in all requirements.txt + fixed the nox ses…

…sion to install nightly on 3.12

.github/workflows/contrib_rerun_py.yml

@@ -68,7 +68,7 @@ jobs:
         # TODO(jleibs): understand why deps can't be installed in the same step as the wheel
         shell: bash
         run: |
-          pip install attrs>=23.1.0 numpy>=1.23 pillow pyarrow==14.0.1 pytest==7.1.2 torch==2.1.0 typing_extensions>=4.5
+          pip install attrs>=23.1.0 numpy>=1.23 pillow pyarrow==14.0.2 pytest==7.1.2 torch==2.1.0 typing_extensions>=4.5
 
       - name: Get version
         id: get-version

.github/workflows/reusable_build_examples.yml

@@ -69,7 +69,7 @@ jobs:
         shell: bash
         run: |
           pixi run pip uninstall rerun-sdk -y
-          pixi run pip install deprecated numpy>=1.23 pyarrow==10.0.1 pytest==7.1.2
+          pixi run pip install deprecated numpy>=1.23 pyarrow==14.0.2 pytest==7.1.2
           pixi run pip install rerun-sdk --no-index --find-links wheel
 
       - name: Verify built wheel version

.github/workflows/reusable_publish_web.yml

@@ -89,7 +89,7 @@ jobs:
           pixi run pip install -r scripts/ci/requirements.txt
           pixi run pip install -r scripts/ci/requirements-web-demo.txt
           pixi run pip uninstall rerun-sdk -y
-          pixi run pip install deprecated numpy>=1.23 pyarrow==10.0.1 pytest==7.1.2
+          pixi run pip install deprecated numpy>=1.23 pyarrow==14.0.2 pytest==7.1.2
           pixi run pip install rerun-sdk --no-index --find-links wheel
 
       - name: Installed wheel version

.github/workflows/reusable_run_notebook.yml

@@ -45,7 +45,7 @@ jobs:
         # TODO(jleibs): pull these deps from pyproject.toml
         shell: bash
         run: |
-          pip install deprecated numpy>=1.23 pillow>=9.5.0 pyarrow==14.0.1 pytest==7.1.2
+          pip install deprecated numpy>=1.23 pillow>=9.5.0 pyarrow==14.0.2 pytest==7.1.2
 
       - name: Install downloaded wheel_artifact
         # Now install the wheel using a specific version and --no-index to guarantee we get the version from

.github/workflows/reusable_test_wheels.yml

@@ -150,7 +150,7 @@ jobs:
         # TODO(jleibs): understand why deps can't be installed in the same step as the wheel
         shell: bash
         run: |
-          pip install attrs>=23.1.0 numpy>=1.23 pillow pyarrow==14.0.1 pytest==7.1.2 torch==2.1.0 typing_extensions>=4.5
+          pip install attrs>=23.1.0 numpy>=1.23 pillow pyarrow==14.0.2 pytest==7.1.2 torch==2.1.0 typing_extensions>=4.5
 
       - name: Get version
         id: get-version

examples/python/controlnet/requirements.txt

@@ -3,6 +3,7 @@ opencv-python
 pillow
 diffusers
 numpy
-torch
+#TODO(#4704): clean that up when pytorch is available for 3.12
+torch ; python_version < "3.12"
 transformers
 rerun-sdk

examples/python/depth_guided_stable_diffusion/requirements.txt

@@ -7,5 +7,6 @@ pillow
 requests>=2.31,<3
 rerun-sdk
 scipy
-torch>=1.13.0
+#TODO(#4704): clean that up when pytorch is available for 3.12
+torch>1.13.0; python_version < "3.12"
 transformers>=4.26.0

examples/python/detect_and_track_objects/requirements.txt

@@ -5,5 +5,6 @@ pillow
 requests>=2.31,<3
 rerun-sdk
 timm==0.9.11
-torch>=1.13.0
+#TODO(#4704): clean that up when pytorch is available for 3.12
+torch>=1.13.0 ; python_version < "3.12"
 transformers

examples/python/llm_embedding_ner/requirements.txt

@@ -1,4 +1,5 @@
 rerun-sdk
-torch
+#TODO(#4704): clean that up when pytorch is available for 3.12
+torch ; python_version < "3.12"
 transformers
 umap-learn

examples/python/segment_anything_model/requirements.txt

@@ -3,6 +3,7 @@ numpy
 opencv-python
 requests>=2.31,<3
 rerun-sdk
-torch
-torchvision
+#TODO(#4704): clean that up when pytorch is available for 3.12
+torch ; python_version < "3.12"
+torchvision ; python_version < "3.12"
 tqdm

noxfile.py

@@ -17,6 +17,12 @@
 def tests(session: nox.Session) -> None:
     """Run the Python test suite"""
     session.install("-r", "rerun_py/requirements-build.txt")
+    # TODO(#4704): clean that up when torch is 3.12 compatible
+    if session.python == "3.12":
+        session.run(
+            "pip", "install", "torch", "torchvision", "--pre", "--index-url", "https://download.pytorch.org/whl/nightly"
+        )
+
     session.install("./rerun_py")
     session.run("just", "py-test", external=True)
 

pixi.toml

@@ -107,7 +107,7 @@ just = ">=1.15.0"
 maturin = ">=0.14,<0.15"
 mypy = "1.4.1"
 pip = ">=23"
-pyarrow = "14.0.1"
+pyarrow = "14.0.2"
 pytest = ">=7"
 python = ">=3.8,<3.12"
 ruff = "0.1.2"

rerun_py/pyproject.toml

@@ -16,13 +16,13 @@ dependencies = [
   "attrs>=23.1.0",
   "numpy>=1.23",
   "pillow",                 # Used for JPEG encoding
-  "pyarrow==14.0.1",
+  "pyarrow==14.0.2",
   "typing_extensions>=4.5", # Used for PEP-702 deprecated decorator
 ]
 description = "The Rerun Logging SDK"
 keywords = ["computer-vision", "logging", "rerun"]
 name = "rerun-sdk"
-requires-python = ">=3.8, <3.12" # TODO(#3699): Support Python 3.12
+requires-python = ">=3.8, <3.13"
 
 [[project.authors]]
 email = "opensource@rerun.io"

rerun_py/requirements-build.txt

@@ -3,4 +3,5 @@ maturin>=1.3.0
 semver
 wheel
 pytest
-#torch  # needed for the test suite
+#TODO(#4704): clean that up when pytorch is available for 3.12
+torch>=2.0.1 ; python_version < "3.12"

rerun_py/requirements-lint.txt

@@ -2,7 +2,8 @@ attrs>=23.1.0  # for mypy to work
 blackdoc==0.3.8
 mypy==1.4.1
 numpy>=1.24 # For mypy plugin
-torch>=2.0.1
+#TODO(#4704): clean that up when pytorch is available for 3.12
+torch>=2.0.1 ; python_version < "3.12"
 pip-check-reqs==2.4.4 # Checks for missing deps in requirements.txt files
 pytest  # For mypy to work
 ruff==0.1.2

scripts/requirements-dev.txt

@@ -8,7 +8,8 @@ cryptography==38.0.4 # for scripts/upload_image.py
 google-cloud-storage==2.9.0 # for scripts/upload_image.py
 PyGithub==1.58.2 # for scripts/ci/generate_pr_summary.py and scripts/ci/update_pr_body.py
 Pillow # for scripts/upload_image.py
-torch
+#TODO(#4704): clean that up when pytorch is available for 3.12
+torch ; python_version < "3.12"
 tqdm
 requests
 gitignore_parser  # handle .gitignore