From 748ce29fef9eb26f098241711539d739756221c7 Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Tue, 27 Dec 2022 11:06:30 -0800
Subject: [PATCH] remove a typing warning

---
 imagen_pytorch/imagen_pytorch.py | 2 +-
 imagen_pytorch/version.py        | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/imagen_pytorch/imagen_pytorch.py b/imagen_pytorch/imagen_pytorch.py
index 4a36fac..dd82aa9 100644
--- 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
index dfcd902..7002b98 100644
--- a/imagen_pytorch/version.py
+++ b/imagen_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.18.5'
+__version__ = '1.18.6'