Fix bug where class probabilities are read as str instead of floats

Author: gustavocidornelas

openlayer/models.py

@@ -1,3 +1,4 @@
+import ast
 import logging
 import os
 import shutil
@@ -359,4 +360,9 @@ def run(self, input_data: pd.DataFrame) -> pd.DataFrame:
             # Read the output data from the csv file
             output_data = pd.read_csv(f"{temp_dir}/output_data.csv")
 
+            # Make the items list of floats (and not strings)
+            output_data["predictions"] = output_data["predictions"].apply(
+                ast.literal_eval
+            )
+
         return output_data