remove a typing warning

lucidrains · Dec 27, 2022 · 748ce29 · 748ce29
1 parent cd8acb0
commit 748ce29
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/imagen_pytorch/imagen_pytorch.py b/imagen_pytorch/imagen_pytorch.py
@@ -1,7 +1,7 @@
 import math
 import copy
 from random import random
-from typing import List, Union
+from beartype.typing import List, Union
 from beartype import beartype
 from tqdm.auto import tqdm
 from functools import partial, wraps

diff --git a/imagen_pytorch/version.py b/imagen_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.18.5'
+__version__ = '1.18.6'