diff --git a/raid/_version.py b/raid/_version.py index a73339b..00ec2dc 100644 --- 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 index f76bb78..da8c84a 100644 --- 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}, ...]