Merge pull request #551 from DagsHub/bug/sample-bug

Bug: Fix ds.sample() crash
DagsHub · Nov 14, 2024 · 69b26fe · 69b26fe
2 parents b1d0b06 + 0bb2cef
commit 69b26fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dagshub/data_engine/model/datasource.py b/dagshub/data_engine/model/datasource.py
@@ -275,7 +275,7 @@ def sample(self, start: Optional[int] = None, end: Optional[int] = None) -> "Que
         if start is not None:
             logger.warning("Starting slices is not implemented for now")
         res = self._source.client.sample(self, end, include_metadata=True)
-        self._download_document_fields(res)
+        res._load_autoload_fields()
         return res
 
     def head(self, size=100, load_documents=True, load_annotations=True) -> "QueryResult":