styling

Author: lfayoux

cohere/client.py

@@ -765,14 +765,10 @@ def list_datasets(self, dataset_type: str = None, limit: int = None, offset: int
             "offset": offset,
         }
         response = self._request(f"{cohere.DATASET_URL}", method="GET", params=param_dict)
-        return (
-            [
-                Dataset.from_dict({"meta": response.get("meta"), **r}, wait_fn=self.wait_for_dataset)
-                for r in response["datasets"]
-            ]
-            if "datasets" in response
-            else []
-        )
+        return [
+            Dataset.from_dict({"meta": response.get("meta"), **r}, wait_fn=self.wait_for_dataset)
+            for r in (response.get("datasets") or [])
+        ]
 
     def delete_dataset(self, id: str) -> None:
         """Deletes your dataset

cohere/client_async.py

@@ -567,14 +567,10 @@ async def list_datasets(
             param_dict["offset"] = offset
 
         response = await self._request(f"{cohere.DATASET_URL}", method="GET", params=param_dict)
-        return (
-            [
-                AsyncDataset.from_dict({"meta": response.get("meta"), **r}, wait_fn=self.wait_for_dataset)
-                for r in response["datasets"]
-            ]
-            if "datasets" in response
-            else []
-        )
+        return [
+            AsyncDataset.from_dict({"meta": response.get("meta"), **r}, wait_fn=self.wait_for_dataset)
+            for r in (response.get("datasets") or [])
+        ]
 
     async def delete_dataset(self, id: str) -> None:
         """Deletes your dataset