Merge pull request #14 from liamdugan/typecheck

Fix `run_detection` passing inputs of wrong type to detector function
liamdugan · Sep 18, 2024 · f644ff8 · f644ff8
2 parents cc80d45 + 644752c
commit f644ff8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/raid/_version.py b/raid/_version.py
@@ -1 +1 @@
-__version__ = "0.0.8"
+__version__ = "0.0.9"
diff --git a/raid/detect.py b/raid/detect.py
@@ -3,7 +3,7 @@ def run_detection(f, df):
     scores_df = df[["id"]].copy()
 
     # Run the detector function on the dataset and put output in score column
-    scores_df["score"] = f(df["generation"])
+    scores_df["score"] = f(df["generation"].tolist())
 
     # Convert scores and ids to dict in 'records' format for seralization
     # e.g. [{'id':'...', 'score':0}, {'id':'...', 'score':1}, ...]