File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
src/lightning/pytorch/trainer Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -548,7 +548,7 @@ def fit(
548
548
def _fit_impl (
549
549
self ,
550
550
model : "pl.LightningModule" ,
551
- compile_kwargs : Optional [Dict [str , Any ]] = None ,
551
+ compile_kwargs : Optional [dict [str , Any ]] = None ,
552
552
train_dataloaders : Optional [Union [TRAIN_DATALOADERS , LightningDataModule ]] = None ,
553
553
val_dataloaders : Optional [EVAL_DATALOADERS ] = None ,
554
554
datamodule : Optional [LightningDataModule ] = None ,
@@ -911,7 +911,7 @@ def _predict_impl(
911
911
def _run (
912
912
self ,
913
913
model : "pl.LightningModule" ,
914
- compile_kwargs : Optional [Dict [str , Any ]] = None ,
914
+ compile_kwargs : Optional [dict [str , Any ]] = None ,
915
915
ckpt_path : Optional [_PATH ] = None ,
916
916
) -> Optional [Union [_EVALUATE_OUTPUT , _PREDICT_OUTPUT ]]:
917
917
if self .state .fn == TrainerFn .FITTING :
You can’t perform that action at this time.
0 commit comments