Allow pytorch 2.0 & pytorch-lightning 2.0 (changes via Copilot Workspace)

.github/workflows/ci.yml

@@ -27,7 +27,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.8", "3.9", "3.10"]
+        python-version: ["3.8", "3.9", "3.10", "3.11"]
     steps:
       - name: Checkout
         uses: actions/checkout@v3

pycave/bayes/gmm/lightning_module.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 import pytorch_lightning as pl
 import torch
-from pytorch_lightning.callbacks import EarlyStopping
+from pytorch_lightning.callbacks.early_stopping import EarlyStopping
 from torchmetrics import MeanMetric
 from pycave.bayes.core import cholesky_precision
 from pycave.utils import NonparametricLightningModule

pycave/bayes/markov_chain/estimator.py

@@ -7,6 +7,7 @@
 from lightkit.data import DataLoader, dataset_from_tensors
 from torch.nn.utils.rnn import PackedSequence
 from torch.utils.data import Dataset
+from pytorch_lightning import Trainer
 from .lightning_module import MarkovChainLightningModule
 from .model import MarkovChainModel, MarkovChainModelConfig
 from .types import collate_sequences, collate_sequences_same_length, SequenceData

pycave/utils/lightning_module.py

@@ -1,6 +1,6 @@
 from abc import ABC, abstractmethod
 from typing import List
-import pytorch_lightning as pl
+import lightning.pytorch as pl
 import torch
 from torch import nn
 

pyproject.toml

@@ -16,8 +16,8 @@ version = "0.0.0"
 lightkit = "^0.5.0"
 numpy = "^1.20.3"
 python = ">=3.8,<3.11"
-pytorch-lightning = "^1.6.0"
-torch = "^1.8.0"
+pytorch-lightning = "^2.0.0"
+torch = "^2.0.0"
 torchmetrics = ">=0.6,<0.12"
 
 [tool.poetry.group.pre-commit.dependencies]