2.5.1 and mypy

Author: t-vi

.github/workflows/docker-build.yml

@@ -108,7 +108,7 @@ jobs:
           - { python_version: "3.11", pytorch_version: "2.2.2", cuda_version: "12.1.0" }
           - { python_version: "3.11", pytorch_version: "2.3.1", cuda_version: "12.1.0" }
           - { python_version: "3.11", pytorch_version: "2.4.1", cuda_version: "12.1.0" }
-          - { python_version: "3.12", pytorch_version: "2.5.0", cuda_version: "12.1.0" }
+          - { python_version: "3.12", pytorch_version: "2.5.1", cuda_version: "12.1.0" }
     steps:
       - uses: actions/checkout@v4
       - uses: docker/setup-buildx-action@v3

requirements/typing.txt

@@ -1,5 +1,5 @@
 mypy==1.11.0
-torch==2.5.0
+torch==2.5.1
 
 types-Markdown
 types-PyYAML

src/lightning/pytorch/core/module.py

@@ -1405,7 +1405,9 @@ def forward(self, x):
         input_sample = self._apply_batch_transfer_handler(input_sample)
 
         file_path = str(file_path) if isinstance(file_path, Path) else file_path
-        torch.onnx.export(self, input_sample, file_path, **kwargs)
+        # PyTorch (2.5) declares file_path to be str | PathLike[Any] | None, but
+        #               BytesIO does work, too.
+        torch.onnx.export(self, input_sample, file_path, **kwargs)  # type: ignore
         self.train(mode)
 
     @torch.no_grad()