add __iter__ to dataset object (#361)

lfayoux · web-flow · commit abfbac487799 · 2024-01-02T13:53:41.000-05:00
* add __iter__ to dataset object
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,4 +1,8 @@
 # Changelog
+## 4.40
+ - [#357] (https://github.com/cohere-ai/cohere-python/pull/361)
+    - Add __iter__ to dataset
+
 ## 4.39
  - [#357] (https://github.com/cohere-ai/cohere-python/pull/357)
     - Embed: add bulk embed support for embed V3
diff --git a/cohere/responses/dataset.py b/cohere/responses/dataset.py
@@ -51,6 +51,9 @@ def __init__(
         self.validation_error = validation_error
         self.validation_warnings = validation_warnings
 
+    def __iter__(self):
+        return self.open()
+
     @classmethod
     def from_dict(cls, data: Dict[str, Any], wait_fn) -> "Dataset":
         download_urls = []
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cohere"
-version = "4.39"
+version = "4.40"
 description = "Python SDK for the Cohere API"
 authors = ["Cohere"]
 readme = "README.md"
