recommended changes

Author: Matt Sokoloff

CHANGELOG.md

@@ -1,5 +1,8 @@
 # Changelog
 
+# Next Release:
+* Added `dataset.export_data_rows()` which returns all `DataRows` for a `Dataset`.
+
 # Version 2.6.0 (2021-06-11)
 ## Fix
 * Upated `create_mask_ndjson` helper function in `image_mal.ipynb` to use the color arg

labelbox/schema/dataset.py

@@ -233,7 +233,7 @@ def data_row_for_external_id(self, external_id):
         return data_rows[0]
 
     def export_data_rows(self, timeout_seconds=120):
-        """ Returns a generator that produces all data rows that are currently attached to this datarow.
+        """ Returns a generator that produces all data rows that are currently attached to this dataset.
 
         Args:
             timeout_seconds (float): Max waiting time, in seconds.
@@ -243,7 +243,7 @@ def export_data_rows(self, timeout_seconds=120):
             LabelboxError: if the export fails or is unable to download within the specified time.
         """
         id_param = "datasetId"
-        query_str = """mutation GetQueuedDataRowsExportUrlPyApi($%s: ID!)
+        query_str = """mutation GetDatasetDataRowsExportUrlPyApi($%s: ID!)
             {exportDatasetDataRows(data:{datasetId: $%s }) {downloadUrl createdAt status}}
         """ % (id_param, id_param)
         sleep_time = 2