Skip to content

Commit 609dde3

Browse files
authored
[SDK-425] Remove unnecessary client argument from ExportTask methods (#1282)
1 parent fd196bd commit 609dde3

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

labelbox/schema/export_task.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -539,32 +539,32 @@ def _get_metadata_header(
539539
res = res["task"]["exportMetadataHeader"]
540540
return _MetadataHeader(**res) if res else None
541541

542-
def get_total_file_size(self, client, task_id: str,
542+
def get_total_file_size(self, task_id: str,
543543
stream_type: StreamType) -> Union[int, None]:
544544
"""Returns the total file size for a specific task."""
545545
if not self._task.status in ["COMPLETE", "FAILED"]:
546546
raise ExportTask.TaskNotReadyException("Task is not ready yet")
547-
header = ExportTask._get_metadata_header(client, task_id, stream_type)
547+
header = ExportTask._get_metadata_header(self._task.client, task_id,
548+
stream_type)
548549
return header.total_size if header else None
549550

550-
def get_total_lines(self, client, task_id: str,
551+
def get_total_lines(self, task_id: str,
551552
stream_type: StreamType) -> Union[int, None]:
552553
"""Returns the total file size for a specific task."""
553554
if not self._task.status in ["COMPLETE", "FAILED"]:
554555
raise ExportTask.TaskNotReadyException("Task is not ready yet")
555-
header = ExportTask._get_metadata_header(client, task_id, stream_type)
556+
header = ExportTask._get_metadata_header(self._task.client, task_id,
557+
stream_type)
556558
return header.total_lines if header else None
557559

558560
def has_result(self) -> bool:
559561
"""Returns whether the task has a result."""
560-
total_size = self.get_total_file_size(self._task.client, self._task.uid,
561-
StreamType.RESULT)
562+
total_size = self.get_total_file_size(self._task.uid, StreamType.RESULT)
562563
return total_size is not None and total_size > 0
563564

564565
def has_errors(self) -> bool:
565566
"""Returns whether the task has errors."""
566-
total_size = self.get_total_file_size(self._task.client, self._task.uid,
567-
StreamType.ERRORS)
567+
total_size = self.get_total_file_size(self._task.uid, StreamType.ERRORS)
568568
return total_size is not None and total_size > 0
569569

570570
@overload

0 commit comments

Comments
 (0)