From 1d47eef6b67a6ad8a88cb9e35f1c900f732eda3e Mon Sep 17 00:00:00 2001
From: Casper da Costa-Luis <tqdm@cdcl.ml>
Date: Thu, 8 Apr 2021 16:08:11 +0100
Subject: [PATCH] basic typing start

- fixes #260
---
 tqdm/std.py | 45 +++++++++++++++++++++++++++++++++++++--------
 1 file changed, 37 insertions(+), 8 deletions(-)

diff --git a/tqdm/std.py b/tqdm/std.py
index 3458162f..378c609e 100644
--- a/tqdm/std.py
+++ b/tqdm/std.py
@@ -18,6 +18,12 @@ from time import time
 from warnings import warn
 from weakref import WeakSet
 
+try:
+    from typing import Any, Iterable, Text, TypeVar, Union
+    _T = TypeVar("_T")
+except ImportError:
+    pass
+
 from ._monitor import TMonitor
 from .utils import (
     CallbackIOWrapper, Comparable, DisableOnWriteError, FormatReplace,
@@ -839,13 +845,36 @@ class tqdm(Comparable):
         elif _Rolling_and_Expanding is not None:
             _Rolling_and_Expanding.progress_apply = inner_generator()
 
-    def __init__(self, iterable=None, desc=None, total=None, leave=True, file=None,
-                 ncols=None, mininterval=0.1, maxinterval=10.0, miniters=None,
-                 ascii=None, disable=False, unit='it', unit_scale=False,
-                 dynamic_ncols=False, smoothing=0.3, bar_format=None, initial=0,
-                 position=None, postfix=None, unit_divisor=1000, write_bytes=None,
-                 lock_args=None, nrows=None, colour=None, delay=0, gui=False,
-                 **kwargs):
+    def __init__(
+        self,
+        iterable=None,  # type: Iterable[_T]
+        desc=None,  # type: Text
+        total=None,  # type: Union[int, float]
+        leave=True,  # type: Union[bool, None]
+        file=None,
+        ncols=None,  # type: int
+        mininterval=0.1,  # type: float
+        maxinterval=10.0,  # type: float
+        miniters=None,  # type: Union[int, float]
+        ascii=None,  # type: Union[bool, Text]
+        disable=False,  # type: Union[bool, None]
+        unit='it',  # type: Text
+        unit_scale=False,  # type: Union[bool, int, float]
+        dynamic_ncols=False,  # type: Union[bool, None]
+        smoothing=0.3,  # type: float
+        bar_format=None,  # type: Text
+        initial=0,  # type: Union[int, float]
+        position=None,  # type: int
+        postfix=None,
+        unit_divisor=1000,  # type: Union[int, float]
+        write_bytes=None,  # type: bool
+        lock_args=None,
+        nrows=None,  # type: int
+        colour=None,  # type: Text
+        delay=0,  # type: float
+        gui=False,  # type: bool
+        **kwargs
+    ):
         """
         Parameters
         ----------
@@ -1516,7 +1545,7 @@ class tqdm(Comparable):
             yield CallbackIOWrapper(t.update, stream, method)
 
 
-def trange(*args, **kwargs):
+def trange(*args, **kwargs):  # type: (*Any, **Any) -> Iterable[int]
     """
     A shortcut for tqdm(xrange(*args), **kwargs).
     On Python3+ range is used instead of xrange.