@@ -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