Add DataFrame guard (#90)

rotationalio · Nov 8, 2023 · a8fbfc7 · a8fbfc7
1 parent 7829e0d
commit a8fbfc7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pyensign/ml/dataframe.py b/pyensign/ml/dataframe.py
@@ -1,6 +1,10 @@
 from enum import Enum
 
-import pandas as pd
+try:
+    import pandas as pd
+except ImportError:
+    pd = None
+    raise ImportError("please `pip install pyensign[ml]` to use PyEnsign DataFrames")
 
 
 class DataFrame(pd.DataFrame):