remove comment and move imports

locopy/database.py

@@ -17,6 +17,8 @@
 """Database Module."""
 
 import time
+import pandas
+import polars
 
 from locopy.errors import CredentialsError, DBError
 from locopy.logger import INFO, get_logger
@@ -223,12 +225,8 @@ def to_dataframe(self, df_type="pandas", size=None):
             return None
 
         if df_type == "pandas":
-            import pandas
-
             return pandas.DataFrame(fetched, columns=columns)
         elif df_type == "polars":
-            import polars
-
             return polars.DataFrame(fetched, schema=columns, orient="row")
 
     def to_dict(self):

locopy/redshift.py

@@ -23,6 +23,10 @@
 import os
 from pathlib import Path
 
+import pandas as pd
+import polars as pl
+import polars.selectors as cs
+
 from locopy.database import Database
 from locopy.errors import DBError, S3CredentialsError
 from locopy.logger import INFO, get_logger
@@ -567,10 +571,6 @@ def insert_dataframe_to_table(
 
 
         """
-        import pandas as pd
-        import polars as pl
-        import polars.selectors as cs
-
         if columns:
             try:
                 dataframe = dataframe[columns]

locopy/snowflake.py

@@ -21,6 +21,9 @@
 """
 
 import os
+import pandas as pd
+import polars as pl
+import polars.selectors as cs
 from pathlib import PurePath
 
 from locopy.database import Database
@@ -423,9 +426,6 @@ def insert_dataframe_to_table(
         metadata: dictionary, optional
             If metadata==None, it will be generated based on data
         """
-        import pandas as pd
-        import polars as pl
-        import polars.selectors as cs
 
         if columns:
             try:
@@ -511,8 +511,6 @@ def to_dataframe(self, df_type="pandas", size=None):
             if df_type == "pandas":
                 return self.cursor.fetch_pandas_all()
             elif df_type == "polars":
-                import polars as pl
-
                 return pl.from_arrow(self.cursor.fetch_arrow_all())
         else:
             return super().to_dataframe(df_type=df_type, size=size)

locopy/utility.py

@@ -25,8 +25,6 @@
 import sys
 import threading
 from collections import OrderedDict
-
-# https://docs.python.org/3/library/functools.html#functools.singledispatch
 from functools import singledispatch
 from itertools import cycle
 from typing import Union