Skip to content

Commit 0b4ce04

Browse files
committed
do not fail on objects in df
1 parent 5b03e9a commit 0b4ce04

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/agentlab/analyze/inspect_results.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,11 @@ def get_constants_and_variables(df: pd.DataFrame, drop_constants: bool = False):
3333
constants = {}
3434
variable_keys = []
3535
for col in df.columns:
36-
if df[col].nunique(dropna=False) == 1:
36+
try:
37+
nuniq = df[col].nunique(dropna=False)
38+
except TypeError:
39+
nuniq = 0 # non hashable types are considered variables
40+
if nuniq == 1:
3741
if isinstance(df[col].iloc[0], np.generic):
3842
val = df[col].iloc[0].item()
3943
else:

0 commit comments

Comments
 (0)