Add timing decorator to query_redshift_w_api

function
slacgismo · Dec 7, 2023 · 6df09ba · 6df09ba
1 parent 56bdf2c
commit 6df09ba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: