fix: minor update after applying suggestions

RobotecAI · Sep 4, 2024 · 9d56dbd · 9d56dbd
1 parent 422e3b2
commit 9d56dbd
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/tests/core/test_rai_cli.py b/tests/core/test_rai_cli.py
@@ -13,7 +13,6 @@
 # limitations under the License.
 
 import os
-import uuid
 from pathlib import Path
 from unittest.mock import MagicMock, patch
 

diff --git a/tests/messages/test_utils.py b/tests/messages/test_utils.py
@@ -16,6 +16,7 @@
 from io import BytesIO
 
 import numpy as np
+import pytest
 from PIL import Image
 
 from rai.messages.utils import preprocess_image
@@ -38,4 +39,3 @@ def decode_image(base64_string: str) -> Image.Image:
 def test_preprocess_image(test_image):
     base64_image = preprocess_image(test_image)
     _ = decode_image(base64_image)  # noqa: F841
-