Fix pandas append

Erotemic · Feb 1, 2024 · 6229032 · 6229032
1 parent 1b88d5e
commit 6229032
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/ibeis/algo/verif/sklearn_utils.py b/ibeis/algo/verif/sklearn_utils.py
@@ -465,8 +465,13 @@ def amean(x, w=None):
     real_id = ['%s' % m for m in target_names]
     confusion_df = pd.DataFrame(confusion, columns=pred_id, index=real_id)
 
-    confusion_df = confusion_df.append(pd.DataFrame(
-        [confusion.sum(axis=0)], columns=pred_id, index=['Σp']))
+    try:
+        confusion_df = confusion_df.append(pd.DataFrame(
+            [confusion.sum(axis=0)], columns=pred_id, index=['Σp']))
+    except Exception:
+        new_row = pd.DataFrame([confusion.sum(axis=0)], columns=pred_id, index=['Σp'])
+        confusion_df = pd.concat([confusion_df, new_row])
+
     confusion_df['Σr'] = np.hstack([confusion.sum(axis=1), [0]])
     confusion_df.index.name = 'real'
     confusion_df.columns.name = 'pred'