lint code

epistoteles · Jun 27, 2024 · 7548514 · 7548514
1 parent 8e20c7d
commit 7548514
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tensorhue/connectors/_pillow.py b/tensorhue/connectors/_pillow.py
@@ -1,7 +1,7 @@
 import numpy as np
 
 
-def _tensorhue_to_numpy_pillow(image, thumbnail, max_size, *args, **kwargs) -> np.ndarray:
+def _tensorhue_to_numpy_pillow(image, thumbnail, max_size) -> np.ndarray:
     try:
         image = image.convert("RGB")
     except Exception as e:

diff --git a/tests/test_connector_pillow.py b/tests/test_connector_pillow.py
@@ -1,5 +1,5 @@
-from PIL import Image
 import os
+from PIL import Image
 import pytest
 from tensorhue.connectors._pillow import _tensorhue_to_numpy_pillow
 

diff --git a/tests/test_viz.py b/tests/test_viz.py
@@ -1,10 +1,10 @@
+import os
 import pytest
 import torch
 import jax.numpy as jnp
 import tensorflow as tf
 import numpy as np
 from PIL import Image
-import os
 from tensorhue.viz import viz
 from tensorhue.connectors._torch import _tensorhue_to_numpy_torch
 from tensorhue.connectors._jax import _tensorhue_to_numpy_jax