Rm pandas dependency

siboehm · Nov 22, 2021 · 26e39d6 · 26e39d6
1 parent fa14013
commit 26e39d6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/lleaves/data_processing.py b/lleaves/data_processing.py
@@ -4,7 +4,6 @@
 from typing import List, Optional
 
 import numpy as np
-import pandas as pd
 
 try:
     from pandas import DataFrame as pd_DataFrame
@@ -16,7 +15,7 @@ class pd_DataFrame:
         pass
 
 
-def _dataframe_to_ndarray(data: pd.DataFrame, pd_traintime_categories: List[List]):
+def _dataframe_to_ndarray(data: pd_DataFrame, pd_traintime_categories: List[List]):
     """
     Converts the given dataframe into a 2D numpy array and converts categorical columns to float.