Merge pull request #134 from haraoka-screen/add_check

Add args checking
cdt15 · Mar 27, 2024 · 0cccccd · 0cccccd
2 parents 8f5b6f5 + 441d087
commit 0cccccd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lingam/utils/__init__.py b/lingam/utils/__init__.py
@@ -1111,6 +1111,9 @@ def calculate_total_effect(adjacency_matrix, from_index, to_index, is_continuous
         max_val=len(adjacency_matrix) - 1,
     )
 
+    if from_index == to_index:
+        raise ValueError("from_index and to_index mustn't be the same.")
+
     if is_continuous is None:
         is_continuous = [True for _ in range(len(adjacency_matrix))]
     else: