fix type error

phelps-sg · Jan 23, 2024 · 8adc95c · 8adc95c
1 parent 23046f6
commit 8adc95c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/llm_cooperation/notebook.py b/llm_cooperation/notebook.py
@@ -21,8 +21,7 @@ def save_table(df: pd.DataFrame, name: str, caption: str) -> pd.DataFrame:
     return df
 
 
-def _repeated_to_long_row(row: pd.DataFrame) -> pd.DataFrame:
-    df = pd.DataFrame(row).transpose()
+def _repeated_to_long_row(df: pd.DataFrame) -> pd.DataFrame:
     choices = df.Choices.values[0]
     n = len(choices)
     user_choices = [c.user.as_int for c in choices]