Bugfixes

ucbepic · Oct 11, 2024 · 0ed8cf8 · 0ed8cf8
1 parent 0dde7f7
commit 0ed8cf8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/docetl/operations/sample.py b/docetl/operations/sample.py
@@ -1,3 +1,4 @@
+import sklearn.model_selection
 from typing import Any, Dict, List, Optional, Tuple
 from .base import BaseOperation
 
@@ -49,4 +50,4 @@ def execute(
                 train_size = samples,
                 random_state = self.config.get("random_state", None),
                 stratify = stratify)
-        return output_data
+        return output_data, 0