Skip to content

Commit e1d72d3

Browse files
committed
fix bugs
1 parent ef2cc90 commit e1d72d3

File tree

2 files changed

+11
-2
lines changed

2 files changed

+11
-2
lines changed

scaleapi/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616

1717

1818
def validate_payload(task_type, kwargs):
19-
allowed_fields = DEFAULT_FIELDS + ALLOWED_FIELDS[task_type]
19+
allowed_fields = DEFAULT_FIELDS | ALLOWED_FIELDS[task_type]
2020
for k in kwargs:
2121
if k not in allowed_fields:
2222
raise ScaleInvalidRequest('Illegal parameter %s for task_type %s'
@@ -88,7 +88,7 @@ def cancel_task(self, task_id):
8888

8989
def tasks(self):
9090
"""Returns a list of all your tasks."""
91-
return [Task(json, self) for json in self._getrequest('tasks')]
91+
return [Task(json, self) for json in self._getrequest('tasks')['docs']]
9292

9393
def create_categorization_task(self, **kwargs):
9494
validate_payload('categorization', kwargs)

scaleapi/tasks.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,5 +11,14 @@ def __getattr__(self, name):
1111
raise AttributeError("'%s' object has no attribute %s"
1212
% (type(self).__name__, name))
1313

14+
def __str__(self):
15+
return 'Task(id=%s)' % self.id
16+
17+
def __repr__(self):
18+
return 'Task(%s)' % self.param_dict
19+
20+
def refresh(self):
21+
self.param_dict = self.client._getrequest('task/%s' % self.id)
22+
1423
def cancel(self):
1524
self.client.cancel_task(self.id)

0 commit comments

Comments
 (0)