update except + optional dependencies

pyproject.toml

@@ -36,7 +36,6 @@ dependencies = [
     'copulas>=0.11.0',
     'tqdm>=4.29',
     'plotly>=5.19.0',
-    'xgboost>=1.1.0'
 ]
 
 [project.urls]
@@ -56,6 +55,7 @@ torch = [
     "torch>=2.2.0;python_version>='3.12'",
 ]
 pomegranate = ['pomegranate>=0.15,<1.0']
+xgboost = ['xgboost>=1.1.0']
 test = [
     'sdmetrics[torch]',
     'pytest>=6.2.5,<7',
@@ -68,7 +68,7 @@ test = [
     'pytest-runner>=2.11.1',
 ]
 dev = [
-    'sdmetrics[test, torch]',
+    'sdmetrics[test, xgboost, torch]',
 
     # general
     'build>=1.0.0,<2',

sdmetrics/single_table/data_augmentation/base.py

@@ -6,13 +6,12 @@
 import pandas as pd
 from sklearn.metrics import confusion_matrix, precision_recall_curve, precision_score, recall_score
 from sklearn.preprocessing import OrdinalEncoder
-from xgboost.core import XGBoostError
 
 from sdmetrics.single_table.data_augmentation._libomp_installation import install_libomp
 
 try:
     from xgboost import XGBClassifier
-except XGBoostError:
+except:
     install_libomp()
     from xgboost import XGBClassifier