Support PyTorch 2.7 (#528)

Author: akihironitta

.github/actions/setup/action.yml

@@ -6,7 +6,7 @@ inputs:
     default: '3.9'
   torch-version:
     required: false
-    default: '2.6'
+    default: '2.7'
   cuda-version:
     required: false
     default: cpu

.github/workflows/testing.yml

@@ -18,24 +18,29 @@ jobs:
       fail-fast: false
       matrix:
         include:
-          - { torch-version: '1.13', python-version: '3.9' }
-          - { torch-version: '1.13', python-version: '3.11' }
-          - { torch-version: '2.0', python-version: '3.9' }
-          - { torch-version: '2.0', python-version: '3.11' }
-          - { torch-version: '2.1', python-version: '3.9' }
-          - { torch-version: '2.1', python-version: '3.11' }
-          - { torch-version: '2.2', python-version: '3.9' }
-          - { torch-version: '2.2', python-version: '3.12' }
-          - { torch-version: '2.3', python-version: '3.9' }
-          - { torch-version: '2.3', python-version: '3.12' }
-          - { torch-version: '2.4', python-version: '3.9' }
-          - { torch-version: '2.4', python-version: '3.12' }
-          - { torch-version: '2.5', python-version: '3.9' }
-          - { torch-version: '2.5', python-version: '3.13' }
-          - { torch-version: '2.6', python-version: '3.9' }
-          - { torch-version: '2.6', python-version: '3.13' }
-          - { torch-version: 'nightly', python-version: '3.9' }
-          - { torch-version: 'nightly', python-version: '3.13' }
+          - {torch-version: '1.13', python-version: '3.9'}
+          - {torch-version: '1.13', python-version: '3.11'}
+          # TODO: Enable these if we have more concurrent runners. PyTorch and
+          # Python are stable enough that we don't need to test against all
+          # versions.
+          # - {torch-version: '2.0', python-version: '3.9'}
+          # - {torch-version: '2.0', python-version: '3.11'}
+          # - {torch-version: '2.1', python-version: '3.9'}
+          # - {torch-version: '2.1', python-version: '3.11'}
+          # - {torch-version: '2.2', python-version: '3.9'}
+          # - {torch-version: '2.2', python-version: '3.12'}
+          # - {torch-version: '2.3', python-version: '3.9'}
+          # - {torch-version: '2.3', python-version: '3.12'}
+          # - {torch-version: '2.4', python-version: '3.9'}
+          # - {torch-version: '2.4', python-version: '3.12'}
+          # - {torch-version: '2.5', python-version: '3.9'}
+          # - {torch-version: '2.5', python-version: '3.13'}
+          # - {torch-version: '2.6', python-version: '3.9'}
+          # - {torch-version: '2.6', python-version: '3.13'}
+          - {torch-version: '2.7', python-version: '3.9'}
+          - {torch-version: '2.7', python-version: '3.13'}
+          - {torch-version: 'nightly', python-version: '3.9'}
+          - {torch-version: 'nightly', python-version: '3.13'}
 
     steps:
       - name: Checkout repository

.pre-commit-config.yaml

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

CHANGELOG.md

@@ -7,6 +7,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
 ### Added
 
+- Added support for PyTorch 2.7 ([#528](https://github.com/pyg-team/pytorch-frame/pull/528))
 - Added a classification example script for TabPFN ([#510](https://github.com/pyg-team/pytorch-frame/pull/510))
 
 ### Changed

docs/requirements.txt

@@ -1,2 +1,2 @@
-https://download.pytorch.org/whl/cpu/torch-2.6.0%2Bcpu-cp39-cp39-linux_x86_64.whl
+https://download.pytorch.org/whl/cpu/torch-2.7.0%2Bcpu-cp39-cp39-manylinux_2_28_x86_64.whl
 git+https://github.com/pyg-team/pyg_sphinx_theme.git