Buffered stream code

Author: Adrian Chang

libs/labelbox/src/labelbox/schema/export_task.py

@@ -850,7 +850,19 @@ def get_buffered_stream(
         self,
         stream_type: StreamType = StreamType.RESULT,
     ) -> Stream:
-        """Returns the result of the task."""
+        """
+        Returns the result of the task.
+
+        Args:
+            stream_type (StreamType, optional): The type of stream to retrieve. Defaults to StreamType.RESULT.
+
+        Returns:
+            Stream: The buffered stream object.
+
+        Raises:
+            ExportTask.ExportTaskException: If the task has failed or is not ready yet.
+            ValueError: If the task does not have the specified stream type.
+        """
         if self._task.status == "FAILED":
             raise ExportTask.ExportTaskException("Task failed")
         if self._task.status != "COMPLETE":
@@ -864,7 +876,7 @@ def get_buffered_stream(
             )
         return BufferedStream(
             _TaskContext(self._task.client, self._task.uid, stream_type,
-                         metadata_header),
+                            metadata_header),
         )
 
     def get_stream(