Fix quickstart example

brainets · Oct 29, 2024 · d2c98bb · d2c98bb
1 parent a85e4e2
commit d2c98bb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/hoi/utils/stats.py b/hoi/utils/stats.py
@@ -250,6 +250,8 @@ def get_nbest_mult(
     import pandas as pd
 
     hoi = np.asarray(hoi).squeeze()
+    if not hoi.ndim:
+        hoi = hoi.reshape(-1)
 
     # get order and multiplets
     if model: