Merge pull request #114 from olaviinha/master

Import Literal from typing_extensions if from typing fails

Author: rromb

taming/data/helper_types.py

@@ -1,8 +1,12 @@
-from typing import Dict, Tuple, Literal, Optional, NamedTuple, Union
-
+from typing import Dict, Tuple, Optional, NamedTuple, Union
 from PIL.Image import Image as pil_image
 from torch import Tensor
 
+try:
+  from typing import Literal
+except ImportError:
+  from typing_extensions import Literal
+
 Image = Union[Tensor, pil_image]
 BoundingBox = Tuple[float, float, float, float]  # x0, y0, w, h
 CropMethodType = Literal['none', 'random', 'center', 'random-2d']