fix

Author: rohitgr7

pytorch_lightning/callbacks/batch_size_finder.py

@@ -21,9 +21,10 @@
 import os
 import uuid
 from copy import deepcopy
-from typing import List, Optional, Tuple, TypedDict, Union
+from typing import List, Optional, Tuple, Union
 
 from torch.utils.data.dataloader import DataLoader
+from typing_extensions import TypedDict
 
 import pytorch_lightning as pl
 from pytorch_lightning.callbacks.base import Callback