resolving MRO

.github/workflows/test.yml

@@ -10,7 +10,8 @@ on:
     branches: [master]
 
 jobs:
-  build:
+  pytest:
+    name: Run pytest
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
@@ -71,7 +72,6 @@ jobs:
         run: poetry run pytest tests
 
       - name: Statistics
-        if: success()
         run: |
           pip install coverage
           coverage report
@@ -89,7 +89,7 @@ jobs:
           fail_ci_if_error: false
 
   docs:
-    name: Test docs build
+    name: Docs build
     runs-on: ubuntu-latest
     defaults:
       run:

docs/requirements.txt

@@ -1,19 +1,19 @@
-sphinx>3.2
+sphinx >3.2
 nbsphinx
 pandoc
 docutils
 pydata-sphinx-theme
-lightning>=2.0.0
+lightning >=2.0.0
 cloudpickle
-torch>=2.0,!=2.0.1
-optuna>=3.1.0
+torch >=2.0,!=2.0.1
+optuna >=3.1.0
 scipy
-pandas>=1.3
-scikit-learn>1.2
+pandas >=1.3
+scikit-learn >1.2
 matplotlib
 statsmodels
 ipython
-nbconvert>=6.3.0
-recommonmark>=0.7.1
-pytorch-optimizer>=2.5.1
-fastapi>0.80
+nbconvert >=6.3.0
+recommonmark >=0.7.1
+pytorch-optimizer >=2.5.1
+fastapi >0.80

pytorch_forecasting/models/temporal_fusion_transformer/tuning.py

@@ -26,7 +26,7 @@
 
 
 # need to inherit from callback for this to work
-class PyTorchLightningPruningCallbackAdjusted(pl.Callback, PyTorchLightningPruningCallback):
+class PyTorchLightningPruningCallbackAdjusted(PyTorchLightningPruningCallback, pl.Callback):
     pass