From 02e3722aba6a69a3e4ac1a88fd9c4f27175fc9c8 Mon Sep 17 00:00:00 2001
From: Hugh Dickinson <hugh.dickinson@open.ac.uk>
Date: Thu, 24 Jun 2021 14:59:34 +0100
Subject: [PATCH] Extra options for read_csv
---
zooniverse/zooniverse.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/zooniverse/zooniverse.py b/zooniverse/zooniverse.py
index 00c52e9..fb31142 100644
--- a/zooniverse/zooniverse.py
+++ b/zooniverse/zooniverse.py
@@ -34,7 +34,7 @@ class zooniverse:
except PanoptesAPIException as e:
return False
- def generate(self, item, wait=False):
+ def generate(self, item, wait=False, **read_csv_args):
print("Generating requested export...")
if wait:
print("\t\tWaiting for generation to complete...")
@@ -57,7 +57,7 @@ class zooniverse:
else:
return None
- def retrieve(self, item, generate=False, wait=False):
+ def retrieve(self, item, generate=False, wait=False, **read_csv_args):
if self.is_available(item) and not generate:
response = self._get_entity(item).get_export(
self._get_item_entry(item, "category"), generate=False, wait=wait
@@ -83,7 +83,7 @@ class zooniverse:
io.BytesIO(response.content),
converters=zooniverse.category_converters[
self._get_item_entry(item, "category")
- ],
+ ], **read_csv_args
)
if wait
else response
--
GitLab