Add exceptions module.

tableone/exceptions.py

@@ -0,0 +1,6 @@
+"""Exceptions and warnings"""
+
+
+class InputError(Exception):
+    """Custom exception for input validation errors."""
+    pass

tableone/preprocessors.py

@@ -1,6 +1,6 @@
 import numpy as np
 
-from tableone.validators import InputError
+from tableone.exceptions import InputError
 
 
 def ensure_list(arg, arg_name):

tableone/statistics.py

@@ -7,7 +7,7 @@
 from statsmodels.stats import multitest
 
 from tableone.modality import hartigan_diptest
-from tableone.validators import InputError
+from tableone.exceptions import InputError
 
 
 class Statistics:

tableone/tables.py

@@ -5,7 +5,7 @@
 import pandas as pd
 
 from tableone.statistics import Statistics
-from tableone.validators import InputError
+from tableone.exceptions import InputError
 
 
 class Tables:

tableone/validators.py

@@ -3,11 +3,6 @@
 import pandas as pd
 
 
-class InputError(Exception):
-    """Custom exception for input validation errors."""
-    pass
-
-
 class DataValidator:
     def __init__(self):
         """Initialize the DataValidator class."""
@@ -169,5 +164,5 @@ def check_columns_exist(self, columns: list, categorical: list, continuous: list
         all_specified = cat_set.union(cont_set)
         if not all_specified.issubset(set(columns)):
             missing = list(all_specified - set(columns))
-            msg = "Specified categorical/continuous columns not found in the DataFrame: f"{missing}"
+            msg = f"Specified categorical/continuous columns not found in the DataFrame: {missing}"
             raise ValueError(msg)