chore: handle legacy mean_df

dattalab · Mar 6, 2023 · 9757d4e · 9757d4e
2 parents 1927d5d + f4bf8c6
commit 9757d4e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/moseq2_viz/model/fingerprint_classifier.py b/moseq2_viz/model/fingerprint_classifier.py
@@ -82,7 +82,10 @@ def create_fingerprint_dataframe(scalar_df, mean_df, stat_type='mean', n_bins=No
 
     if len(vel_cols) > 0:
         scalar_df[vel_cols] *= 30
-        mean_df[vel_cols_stats] *=30
+        try:
+            mean_df[vel_cols_stats] *=30
+        except KeyError:
+            mean_df[vel_cols] *=30
 
     # pivot mean_df to be groupby x syllable
     syll_summary = mean_df.pivot_table(index=groupby_list, values='usage', columns='syllable')