pydoc types

pytube/streams.py

@@ -147,9 +147,10 @@ class Stream(object):
     def download(self, output_path=None):
         """Write the media stream to disk.
 
-        :param str output_path:
+        :param output_path:
             (optional) Output path for writing media file. If one is not
             specified, defaults to the current working directory.
+        :type output_path: str or None
 
         """
         # TODO(nficano): allow a filename to specified.
@@ -179,7 +180,7 @@ class Stream(object):
 
         :param str chunk:
             Segment of media file binary data, not yet written to disk.
-        :param :class:`~_io.BufferedWriter` file_handle:
+        :param BufferedWriter file_handle:
             The file handle where the media is being written to.
         :param int bytes_remaining:
             The delta between the total file size in bytes and amount already
@@ -203,8 +204,10 @@ class Stream(object):
 
     def on_complete(self, file_handle):
         """On download complete handler function.
-        :param :class:`~_io.BufferedWriter` file_handle:
+        :param file_handle:
             The file handle where the media is being written to.
+        :type file_handle:
+            BufferedWriter
 
         """
         logger.debug('download finished')