Allowed for different combinations of scalar and list args to Perfect…

…Intervention.
djpasseyjr · Sep 19, 2024 · 65c5fe5 · 65c5fe5
1 parent 35757ea
commit 65c5fe5
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/interfere/interventions.py b/interfere/interventions.py
@@ -150,11 +150,11 @@ def __init__(
             intervention([10.0, 4.0, 4.0], 0) == [1.6, 4.0, 0.7] # (True.)
         """
         # The case where indexs and constants are floats or ints
-        if isinstance(intervened_idxs, int) and isinstance(constants, (int, float)):
-            i = intervened_idxs
-            c = float(constants)
-            intervened_idxs = [intervened_idxs]
-            constants = [c]
+        if isinstance(intervened_idxs, (int, float)):
+            intervened_idxs = [int(intervened_idxs)]
+
+        if isinstance(constants, (int, float)):
+            constants = [float(constants)]
 
         if len(constants) != len(intervened_idxs):
             raise ValueError(