update unit workflow

Author: R-Palazzo

.github/workflows/integration.yml

@@ -35,7 +35,7 @@ jobs:
     - name: Install dependencies
       run: |
           python -m pip install --upgrade pip
-          python -m pip install invoke .[pomegranate, xgboost, test]
+          python -m pip install invoke .[pomegranate,xgboost,test]
     - name: Run integration tests
       run: invoke integration
 

.github/workflows/minimum.yml

@@ -35,6 +35,6 @@ jobs:
     - name: Install dependencies
       run: |
           python -m pip install --upgrade pip
-          python -m pip install invoke .[test]
+          python -m pip install invoke .[test,xgboost]
     - name: Test with minimum versions
       run: invoke minimum

.github/workflows/unit.yml

@@ -27,6 +27,11 @@ jobs:
       uses: actions/setup-python@v5
       with:
         python-version: ${{ matrix.python-version }}
+    - name: Install libomp (macOS only)
+      if: matrix.os == 'macos-latest'
+      run: |
+          brew install libomp
+          echo 'export DYLD_LIBRARY_PATH=$(brew --prefix libomp)/lib:$DYLD_LIBRARY_PATH' >> $GITHUB_ENV
     - name: Install dependencies
       run: |
           python -m pip install --upgrade pip

sdmetrics/single_table/data_augmentation/_libomp_installation.py

@@ -6,15 +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 import XGBClassifier
 
 from sdmetrics.goal import Goal
 from sdmetrics.single_table.base import SingleTableMetric
-from sdmetrics.single_table.data_augmentation._libomp_installation import install_libomp
 from sdmetrics.single_table.data_augmentation.utils import _validate_inputs
 
-install_libomp()
-from xgboost import XGBClassifier  # noqa
-
 METRIC_NAME_TO_METHOD = {'recall': recall_score, 'precision': precision_score}