diff --git a/solardatatools/dataio.py b/solardatatools/dataio.py index 60a46a49..bf180a23 100644 --- a/solardatatools/dataio.py +++ b/solardatatools/dataio.py @@ -288,7 +288,7 @@ def decompress_data_to_dataframe(encoded_data): return df - def timing(verbose: bool = True): + def timing(verbose: bool = True) -> Callable: def decorator(func: Callable): @wraps(func) def wrapper(*args, **kwargs): @@ -304,7 +304,7 @@ def wrapper(*args, **kwargs): return decorator - # @timing(verbose) + @timing(verbose) def query_redshift_w_api( params: QueryParams, page: int, is_batch: bool = False ) -> requests.Response: