Support PyTorch 2.11

.github/actions/setup/action.yml

@@ -6,7 +6,7 @@ inputs:
     default: '3.10'
   torch-version:
     required: false
-    default: '2.10'
+    default: '2.11'
   cuda-version:
     required: false
     default: cpu

.github/workflows/testing.yml

@@ -20,8 +20,8 @@ jobs:
         include:
           - {torch-version: '2.7', python-version: '3.10'}
           - {torch-version: '2.7', python-version: '3.13'}
-          - {torch-version: '2.10', python-version: '3.10'}
-          - {torch-version: '2.10', python-version: '3.14'}
+          - {torch-version: '2.11', python-version: '3.10'}
+          - {torch-version: '2.11', python-version: '3.14'}
           - {torch-version: 'nightly', python-version: '3.10'}
           - {torch-version: 'nightly', python-version: '3.14'}
 

.pre-commit-config.yaml

@@ -76,7 +76,7 @@ repos:
     hooks:
       - id: mypy
         name: Check types
-        additional_dependencies: [torch==2.9.*]
+        additional_dependencies: [torch==2.11.*]
         exclude: "^test/|^examples/|^benchmark/"
 
   - repo: https://github.com/executablebooks/mdformat

CHANGELOG.md

@@ -7,6 +7,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/).
 
 ### Added
 
+- Added support for PyTorch 2.11
 - Added support for PyTorch 2.10 and Python 3.14 ([#590](https://github.com/pyg-team/pytorch-frame/pull/590))
 
 ### Changed

docs/requirements.txt

@@ -1,2 +1,2 @@
-https://download.pytorch.org/whl/cpu/torch-2.9.0%2Bcpu-cp310-cp310-manylinux_2_28_x86_64.whl
+https://download.pytorch.org/whl/cpu/torch-2.11.0%2Bcpu-cp310-cp310-manylinux_2_28_x86_64.whl
 git+https://github.com/pyg-team/pyg_sphinx_theme.git

test/nn/models/test_compile.py

@@ -14,7 +14,7 @@
 from torch_frame.testing import withPackage
 
 
-@withPackage("torch>=2.11.0")
+@withPackage("torch>=2.12.0")
 @pytest.mark.parametrize(
     "model_cls, model_kwargs, stypes, expected_graph_breaks",
     [