From c134edf4b3eb08afaa48b75e4da8290b21e9751d Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sat, 25 Nov 2023 15:28:03 -0600
Subject: [PATCH] sonar lint

---
 api/tests/chain/test_upscale_outpaint.py | 4 +---
 api/tests/chain/test_upscale_swinir.py   | 4 +---
 api/tests/worker/test_worker.py          | 2 +-
 3 files changed, 3 insertions(+), 7 deletions(-)

diff --git a/api/tests/chain/test_upscale_outpaint.py b/api/tests/chain/test_upscale_outpaint.py
index 0e524014..261a8d45 100644
--- a/api/tests/chain/test_upscale_outpaint.py
+++ b/api/tests/chain/test_upscale_outpaint.py
@@ -28,9 +28,7 @@ class UpscaleOutpaintStageTests(unittest.TestCase):
                 3,
                 0.1,
             ),
-            ServerContext(
-                # model_path="../models",
-            ),
+            ServerContext(),
             None,
             ImageParams(
                 "../models/stable-diffusion-onnx-v1-inpainting",
diff --git a/api/tests/chain/test_upscale_swinir.py b/api/tests/chain/test_upscale_swinir.py
index e2bf69fe..dfa9676e 100644
--- a/api/tests/chain/test_upscale_swinir.py
+++ b/api/tests/chain/test_upscale_swinir.py
@@ -28,9 +28,7 @@ class UpscaleSwinIRStageTests(unittest.TestCase):
                 3,
                 0.1,
             ),
-            ServerContext(
-                # model_path="../models",
-            ),
+            ServerContext(),
             None,
             None,
             sources,
diff --git a/api/tests/worker/test_worker.py b/api/tests/worker/test_worker.py
index 553094ad..6365fac9 100644
--- a/api/tests/worker/test_worker.py
+++ b/api/tests/worker/test_worker.py
@@ -18,7 +18,7 @@ from tests.helpers import test_device
 
 
 def main_memory(_worker):
-    raise Exception(MEMORY_ERRORS[0])
+    raise MemoryError(MEMORY_ERRORS[0])
 
 
 def main_retry(_worker):