bug fix

verbal-autopsy-software · Jul 24, 2023 · c2452a9 · c2452a9
1 parent 8a4dd9e
commit c2452a9
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/interva/interva5.py b/interva/interva5.py
@@ -684,8 +684,10 @@ def get_csmf(self, top: int = 10,
         # for future compatibility with non-standard input
         causenames = causeindex = []
         for i in range(va.shape[0]):
-            if va.loc[i, "WHOLEPROB"] is not None:
-                causenames = va.loc[i, "WHOLEPROB"].index
+            # if va.loc[i, "WHOLEPROB"] is not None:
+            #     causenames = va.loc[i, "WHOLEPROB"].index
+            if va.iloc[i]["WHOLEPROB"] is not None:
+                causenames = va.iloc[i]["WHOLEPROB"].index
                 causeindex = [x for x in range(len(causenames))]
                 break
         include_probAC = False
@@ -870,7 +872,8 @@ def get_indiv_prob(self, top: int = 0,
                     column_names.append(prob)
 
         for indiv in range(num_indiv):
-            wholeprob = VA5.loc[indiv, "WHOLEPROB"]
+            # wholeprob = VA5.loc[indiv, "WHOLEPROB"]
+            wholeprob = VA5.iloc[indiv]["WHOLEPROB"]
             prob_B = wholeprob.iloc[3:64].copy()
 
             if top == 0 or top is None: