update

sktime · Oct 9, 2024 · 0d9d4dc · 0d9d4dc
1 parent 622d979
commit 0d9d4dc
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/pytorch_forecasting/models/temporal_fusion_transformer/tuning.py b/pytorch_forecasting/models/temporal_fusion_transformer/tuning.py
@@ -22,11 +22,6 @@
 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,