[ENH] Changed to isinstance

dutta-alankar · Apr 28, 2024 · ea3dfc6 · ea3dfc6
1 parent 417daaa
commit ea3dfc6
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/.tokeignore b/.tokeignore
diff --git a/astro_plasma/core/functions.py b/astro_plasma/core/functions.py
@@ -30,9 +30,9 @@ def cooling_approx(temperature, metallicity):
         Plasma cooling function normalized by nH^2.
 
     """
-    if type(temperature) == list:
+    if isinstance(temperature, list):
         temperature = np.array(temperature)
-    if type(metallicity) == list:
+    if isinstance(metallicity, list):
         metallicity = np.array(metallicity)
     file_path = os.path.realpath(__file__)
     dir_loc = os.path.split(file_path)[:-1]

diff --git a/astro_plasma/core/utils.py b/astro_plasma/core/utils.py
@@ -66,7 +66,7 @@ class AtmElement(Enum):
 
     @staticmethod
     def parse(predicate: Union[str, int]) -> "AtmElement":
-        idx = int(type(predicate) == int)
+        idx = int(isinstance(predicate, int))
         error_label = "atomic number" if idx == 1 else "symbol"
 
         for element in AtmElement: