Merge pull request #2 from hugovk/lazy-pandas

Lazily import pandas to speedup non-pandas use
thombashi · Feb 17, 2023 · b162600 · b162600
2 parents 8b6369c + c50adb3
commit b162600
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/tabledata/_core.py b/tabledata/_core.py
@@ -17,14 +17,6 @@
 from ._logger import logger
 
 
-try:
-    from pandas import DataFrame
-
-    INSTALLED_PANDAS = True
-except ImportError:
-    INSTALLED_PANDAS = False
-
-
 class TableData:
     """
     Class to represent a table data structure.
@@ -428,7 +420,9 @@ def as_dataframe(self) -> "DataFrame":
             - `pandas <https://pandas.pydata.org/>`__
         """
 
-        if not INSTALLED_PANDAS:
+        try:
+            from pandas import DataFrame
+        except ImportError:
             raise RuntimeError("required 'pandas' package to execute as_dataframe method")
 
         dataframe = DataFrame(self.value_matrix)