Merge pull request #626 from Julat300/ContrastDir

Made the contrast_curve function more robust to classes with inherited attributes
vortex-exoplanet · Dec 18, 2023 · 5785214 · 5785214
2 parents 6028e7e + 20da21c
commit 5785214
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/vip_hci/metrics/contrcurve.py b/vip_hci/metrics/contrcurve.py
@@ -730,7 +730,7 @@ def throughput(
         mod = algo.__module__[:idx]
         tmp = __import__(mod, fromlist=[algo_name.upper()+'_Params'])
         algo_params = getattr(tmp, algo_name.upper()+'_Params')
-        argl = [attr for attr in vars(algo_params)]
+        argl = [attr for attr in dir(algo_params)]
         if "cube" in argl and "angle_list" in argl and "verbose" in argl:
             # (ii) a VIP postproc algorithm [OK]
             pass
@@ -870,7 +870,7 @@ def throughput(
                     tmp = __import__(
                         mod, fromlist=[algo_name.upper()+'_Params'])
                     algo_params = getattr(tmp, algo_name.upper()+'_Params')
-                    arg = [attr for attr in vars(algo_params)]
+                    arg = [attr for attr in dir(algo_params)]
                     if "cube" in arg and "angle_list" in arg and "verbose" in arg:
                         # (ii) a VIP postproc algorithm [OK]
                         pass
@@ -996,7 +996,7 @@ def throughput(
                     tmp = __import__(
                         mod, fromlist=[algo_name.upper()+'_Params'])
                     algo_params = getattr(tmp, algo_name.upper()+'_Params')
-                    arg = [attr for attr in vars(algo_params)]
+                    arg = [attr for attr in dir(algo_params)]
                     if "cube" in arg and "angle_list" in arg and "verbose" in arg:
                         # (ii) a VIP postproc algorithm [OK]
                         pass