ruff check fixes

spacetelescope · Sep 17, 2024 · ef36b3f · ef36b3f
1 parent 25c6092
commit ef36b3f
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 8 deletions.
diff --git a/spacekit/analyzer/explore.py b/spacekit/analyzer/explore.py
@@ -980,7 +980,7 @@ def atomic_vector_plotter(
             x_units = x_units
 
         # Scatter Plot
-        if type(signal) == np.array:
+        if isinstance(signal, np.array):
             series_index = list(range(len(signal)))
 
             converted_array = pd.Series(signal.ravel(), index=series_index)

diff --git a/spacekit/extractor/load.py b/spacekit/extractor/load.py
@@ -377,7 +377,7 @@ def split_arrays(self, data, t=0.6, v=0.85):
         arrays
             split sampled arrays
         """
-        if type(data) == pd.DataFrame:
+        if isinstance(data, pd.DataFrame):
             sample = data.sample(frac=1)
         else:
             sample = data

diff --git a/spacekit/extractor/scrape.py b/spacekit/extractor/scrape.py
@@ -9,6 +9,7 @@
 import csv
 from zipfile import ZipFile
 from astropy.io import fits, ascii
+from astropy.table import Table
 from botocore.config import Config
 from decimal import Decimal
 from boto3.dynamodb.conditions import Attr
@@ -1428,18 +1429,15 @@ def make_dataframe_line(self, json_filename_list):
                 ingest_key = fd_key.replace(" ", "_")
                 key_suffix = ingest_key.split(".")[-1]
                 if key_suffix not in ["data", "unit", "format", "dtype"]:
-                    if (
-                        str(type(json_data_item))
-                        == "<class 'astropy.table.table.Table'>"
-                    ):
+                    if isinstance(json_data_item, Table):
                         for coltitle in json_data_item.colnames:
                             ingest_value = json_data_item[coltitle].tolist()
                             id_key = title_suffix + ingest_key + "." + coltitle
                             ingest_dict["data"][id_key] = [ingest_value]
                     else:
                         ingest_value = json_data_item
                         id_key = title_suffix + ingest_key
-                        if str(type(ingest_value)) == "<class 'list'>":
+                        if isinstance(ingest_value, list):
                             ingest_dict["data"][id_key] = [ingest_value]
                         else:
                             ingest_dict["data"][id_key] = ingest_value

diff --git a/spacekit/preprocessor/transform.py b/spacekit/preprocessor/transform.py
@@ -775,7 +775,7 @@ def normalize_training_images(X_tr, X_ts, X_vl=None):
 
 
 def array_to_tensor(arr, reshape=False, shape=(-1, 1)):
-    if type(arr) == tf.Tensor:
+    if isinstance(arr, tf.Tensor):
         return arr
     if reshape is True:
         arr = arr.reshape(shape[0], shape[1])