make release-tag: Merge branch 'main' into stable

Author: amontanez24

.github/workflows/install.yml

@@ -0,0 +1,38 @@
+name: Install Tests
+on:
+  pull_request:
+    types: [opened, synchronize]
+  push:
+    branches:
+      - main
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  install:
+    name: ${{ matrix.python_version }} install
+    strategy:
+      fail-fast: true
+      matrix:
+        python_version: ["3.8", "3.13"]
+    runs-on: ubuntu-latest
+    steps:
+      - name: Set up python ${{ matrix.python_version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python_version }}
+      - uses: actions/checkout@v4
+      - name: Build package
+        run: |
+          make package
+      - name: Install package
+        run: |
+          python -m pip install "unpacked_sdist/."
+      - name: Test by importing packages
+        run: |
+          python -c "import sdmetrics"
+      - name: Check package conflicts
+        run: |
+          python -m pip check

HISTORY.md

@@ -1,5 +1,11 @@
 # History
 
+## v0.20.1 - 2025-04-14
+
+### Bugs Fixed
+
+* Importing sdmetrics crashes because of torch dependency - Issue [#764](https://github.com/sdv-dev/SDMetrics/issues/764) by @amontanez24
+
 ## v0.20.0 - 2025-04-11
 
 ### New Features
@@ -12,10 +18,10 @@
 
 ### Bugs Fixed
 
-* Visualization `get_column_plot` crashes if I have a column of all null values - Issue [#757](https://github.com/sdv-dev/SDMetrics/issues/757) by @npatki
+* Visualization `get_column_plot` crashes if I have a column of all null values - Issue [#757](https://github.com/sdv-dev/SDMetrics/issues/757) by @lajohn4747
 * The `DCRBaselineProtection` metric is not creating the correct size of random data - Issue [#743](https://github.com/sdv-dev/SDMetrics/issues/743) by @lajohn4747
 * The `DCRBaselineProtection` metric is not producing the expected score - Issue [#742](https://github.com/sdv-dev/SDMetrics/issues/742) by @lajohn4747
-* The DCRBaselineProtection metric crashes when the distance between random data and real data is 0 - Issue [#738](https://github.com/sdv-dev/SDMetrics/issues/738) by @npatki
+* The DCRBaselineProtection metric crashes when the distance between random data and real data is 0 - Issue [#738](https://github.com/sdv-dev/SDMetrics/issues/738) by @lajohn4747
 * DCROverfitting and DCRBaseline metrics produce too many warnings about missing columns. - Issue [#737](https://github.com/sdv-dev/SDMetrics/issues/737) by @lajohn4747
 * Wrong default values for the `classifier` parameter of the `BinaryClassifierPrecisionEfficacy` - Issue [#734](https://github.com/sdv-dev/SDMetrics/issues/734) by @R-Palazzo
 

Makefile

@@ -232,3 +232,23 @@ release-minor: check-release bumpversion-minor release
 
 .PHONY: release-major
 release-major: check-release bumpversion-major release
+
+# Packaging Targets
+.PHONY: upgradepip
+upgradepip:
+	python -m pip install --upgrade pip
+
+.PHONY: upgradebuild
+upgradebuild:
+	python -m pip install --upgrade build
+
+.PHONY: upgradesetuptools
+upgradesetuptools:
+	python -m pip install --upgrade setuptools
+
+.PHONY: package
+package: upgradepip upgradebuild upgradesetuptools
+	python -m build ; \
+	$(eval VERSION=$(shell python -c 'import setuptools; setuptools.setup()' --version))
+	tar -zxvf "dist/sdmetrics-${VERSION}.tar.gz"
+	mv "sdmetrics-${VERSION}" unpacked_sdist

pyproject.toml

@@ -141,7 +141,7 @@ convention = 'google'
 add-ignore = ['D107', 'D407', 'D417']
 
 [tool.bumpversion]
-current_version = "0.20.0"
+current_version = "0.20.1.dev1"
 parse = '(?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)(\.(?P<release>[a-z]+)(?P<candidate>\d+))?'
 serialize = [
     '{major}.{minor}.{patch}.{release}{candidate}',

sdmetrics/__init__.py

@@ -4,7 +4,7 @@
 
 __author__ = 'MIT Data To AI Lab'
 __email__ = '[email protected]'
-__version__ = '0.20.0'
+__version__ = '0.20.1.dev1'
 
 import sys
 import warnings as python_warnings

sdmetrics/single_table/bayesian_network.py

@@ -4,11 +4,15 @@
 
 import numpy as np
 import pandas as pd
-import torch
 
 from sdmetrics.goal import Goal
 from sdmetrics.single_table.base import SingleTableMetric
 
+try:
+    import torch
+except ModuleNotFoundError:
+    torch = None
+
 LOGGER = logging.getLogger(__name__)
 
 
@@ -19,6 +23,9 @@ class BNLikelihoodBase(SingleTableMetric):
     def _likelihoods(cls, real_data, synthetic_data, metadata=None, structure=None):
         try:
             from pomegranate.bayesian_network import BayesianNetwork
+
+            if torch is None:
+                raise ImportError
         except ImportError:
             raise ImportError(
                 'Please install pomegranate with `pip install sdmetrics[pomegranate]`.'

static_code_analysis.txt

@@ -1,4 +1,4 @@
-Run started:2025-04-10 20:35:13.475556
+Run started:2025-04-14 17:22:41.156789
 
 Test results:
 >> Issue: [B101:assert_used] Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.

tests/unit/single_table/test_bayesian_network.py

@@ -46,7 +46,20 @@ def metadata():
 class TestBNLikelihood:
     @patch.dict('sys.modules', {'pomegranate.bayesian_network': None})
     def test_compute_error(self):
-        """Test that an `ImportError` is raised."""
+        """Test that an `ImportError` is raised when pomegranate isn't installed."""
+        # Setup
+        metric = BNLikelihood()
+
+        # Run and Assert
+        expected_message = re.escape(
+            'Please install pomegranate with `pip install sdmetrics[pomegranate]`.'
+        )
+        with pytest.raises(ImportError, match=expected_message):
+            metric.compute(Mock(), Mock())
+
+    @patch.dict('sys.modules', {'torch': None})
+    def test_compute_error_torch_is_none(self):
+        """Test that an `ImportError` is raised when torch isn't installed."""
         # Setup
         metric = BNLikelihood()