From e72ffd6d8b7c2b4ddcab986849e050d0ac558e26 Mon Sep 17 00:00:00 2001
From: 0xMRTT <0xMRTT@proton.me>
Date: Wed, 21 Jun 2023 10:52:48 +0200
Subject: [PATCH] fix: type

---
 matrixai/__init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/matrixai/__init__.py b/matrixai/__init__.py
index bf083ae..0c1af93 100644
--- a/matrixai/__init__.py
+++ b/matrixai/__init__.py
@@ -578,7 +578,7 @@ class AsyncImagine:
 
     async def sdprem(self, prompt: str, negative: str = None, priority: str = None, steps: str = None,
                     high_res_results: str = None, style: Style = Style.IMAGINE_V1, seed: str = None,
-                    ratio: Ratio = Ratio.RATIO_1X1, cfg: float = 9.5) -> bytes | None:
+                    ratio: Ratio = Ratio.RATIO_1X1, cfg: float = 9.5) -> bytes:
         """Generates AI Art."""
         try:
             validated_cfg = validate_cfg(cfg)
@@ -615,7 +615,7 @@ class AsyncImagine:
                 else:
                     return None
 
-    async def upscale(self, image: bytes) -> bytes | None:
+    async def upscale(self, image: bytes) -> bytes:
         """Upscales the image."""
         try:
             async with self.session.post(