backwards compatibility

voxel51 · Jan 10, 2024 · 5a9b163 · 5a9b163
1 parent 92ffd9f
commit 5a9b163
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/fiftyone/brain/internal/core/sklearn.py b/fiftyone/brain/internal/core/sklearn.py
@@ -123,16 +123,15 @@ def __init__(
         label_ids=None,
         backend=None,
     ):
-        dataset = samples._dataset
         embeddings, sample_ids, label_ids = self._parse_data(
-            dataset,
+            samples,
             config,
             embeddings=embeddings,
             sample_ids=sample_ids,
             label_ids=label_ids,
         )
 
-        self._dataset = dataset
+        self._dataset = samples._dataset
         self._embeddings = embeddings
         self._sample_ids = sample_ids
         self._label_ids = label_ids
@@ -667,7 +666,7 @@ def _parse_data(
     ):
         if embeddings is None:
             embeddings, sample_ids, label_ids = fbu.get_embeddings(
-                samples,
+                samples._dataset,
                 patches_field=config.patches_field,
                 embeddings_field=config.embeddings_field,
             )