fix: typo in pandas error

dreamquark-ai · Nov 12, 2021 · 5ac5583 · 5ac5583
1 parent d725101
commit 5ac5583
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_tabnet/utils.py b/pytorch_tabnet/utils.py
@@ -348,7 +348,7 @@ def check_input(X):
  """
  if isinstance(X, (pd.DataFrame, pd.Series)):
  err_message = "Pandas DataFrame are not supported: apply X.values when calling fit"
- raise(ValueError, err_message)
+ raise TypeError(err_message)
  check_array(X)