hotfix: dataset.df

etalab-ia · Dec 20, 2024 · bab2467 · bab2467
1 parent ee96a21
commit bab2467
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/schemas.py b/api/schemas.py
@@ -251,7 +251,7 @@ def to_table_init(self, db: Session) -> dict:
             dataset = models.Dataset(**obj["dataset"])
         obj["dataset"] = dataset
         if dataset.has_output:
-            df = pd.read_json(StringIO(dataset))
+            df = pd.read_json(StringIO(dataset.df))
             obj["num_try"] = dataset.size
             obj["num_success"] = df["output"].count()