resolving MRO

Author: Borda

pytorch_forecasting/models/temporal_fusion_transformer/tuning.py

@@ -22,6 +22,11 @@
 optuna_logger = logging.getLogger("optuna")
 
 
+# need to inherit from callback for this to work
+class PyTorchLightningPruningCallbackAdjusted(PyTorchLightningPruningCallback, pl.Callback):
+    pass
+
+
 def optimize_hyperparameters(
     train_dataloaders: DataLoader,
     val_dataloaders: DataLoader,