sdv 1.37.1.dev0__tar.gz → 1.37.2.dev0__tar.gz

{sdv-1.37.1.dev0/sdv.egg-info → sdv-1.37.2.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sdv
-Version: 1.37.1.dev0
+Version: 1.37.2.dev0
 Summary: Generate synthetic data for single table, multi table and sequential data
 Author-email: "DataCebo, Inc." <info@sdv.dev>
 License-Expression: BUSL-1.1

{sdv-1.37.1.dev0 → sdv-1.37.2.dev0}/pyproject.toml

@@ -149,7 +149,7 @@ namespaces = false
 version = {attr = 'sdv.__version__'}
 
 [tool.bumpversion]
-current_version = "1.37.1.dev0"
+current_version = "1.37.2.dev0"
 parse = '(?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)(\.(?P<release>[a-z]+)(?P<candidate>\d+))?'
 serialize = [
     '{major}.{minor}.{patch}.{release}{candidate}',

{sdv-1.37.1.dev0 → sdv-1.37.2.dev0}/sdv/__init__.py

@@ -6,7 +6,7 @@
 
 __author__ = 'DataCebo, Inc.'
 __email__ = 'info@sdv.dev'
-__version__ = '1.37.1.dev0'
+__version__ = '1.37.2.dev0'
 
 
 import sys

{sdv-1.37.1.dev0 → sdv-1.37.2.dev0}/sdv/cag/__init__.py

@@ -7,6 +7,7 @@ from sdv.cag.range import Range
 from sdv.cag.one_hot_encoding import OneHotEncoding
 from sdv.cag.programmable_constraint import (
     ProgrammableConstraint,
+    SingleTableProgrammableConstraint,
 )
 
 __all__ = (
@@ -16,4 +17,5 @@ __all__ = (
     'Range',
     'OneHotEncoding',
     'ProgrammableConstraint',
+    'SingleTableProgrammableConstraint',
 )

{sdv-1.37.1.dev0 → sdv-1.37.2.dev0}/sdv/cag/programmable_constraint.py

@@ -3,13 +3,14 @@
 import inspect
 from copy import deepcopy
 
+from sdv.cag._errors import ConstraintNotMetError
 from sdv.cag.base import BaseConstraint
 
 
 class ProgrammableConstraint:
     """Base class to create programmable constraints."""
 
-    _is_single_table = True
+    _is_single_table = False
 
     @classmethod
     def load_constraint_from_dict(cls, parameters):
@@ -130,6 +131,12 @@ class ProgrammableConstraint:
         return synthetic_data
 
 
+class SingleTableProgrammableConstraint(ProgrammableConstraint):
+    """Single table variant of the base programmable constraint class."""
+
+    _is_single_table = True
+
+
 class ProgrammableConstraintHarness(BaseConstraint):
     """Constraint interface class for programmable constraints."""
 
@@ -139,6 +146,10 @@ class ProgrammableConstraintHarness(BaseConstraint):
         self.table_name = getattr(self.programmable_constraint, 'table_name', None)
         self._is_single_table = self.programmable_constraint._is_single_table
 
+    @property
+    def _is_programmable_single_table_constraint(self):
+        return isinstance(self.programmable_constraint, SingleTableProgrammableConstraint)
+
     def get_constraint_dict(self):
         """Return the constraint as a serialiazable dict.
 
@@ -170,9 +181,21 @@ class ProgrammableConstraintHarness(BaseConstraint):
         }
 
     def _validate_constraint_with_metadata(self, metadata):
+        if self._is_programmable_single_table_constraint and len(metadata.tables) != 1:
+            if getattr(self.programmable_constraint, 'table_name', None) is None:
+                raise ConstraintNotMetError(
+                    'SingleTableProgrammableConstraint cannot be used with multi-table metadata '
+                    'if the `table_name` attribute has not been set. Please set the table name '
+                    'attribute to the target table, or use the ProgrammableContraint '
+                    'base class instead.'
+                )
+
         self.programmable_constraint.validate(metadata)
 
     def _validate_constraint_with_data(self, data, metadata):
+        if self._is_programmable_single_table_constraint:
+            data = data[self._get_single_table_name(metadata)]
+
         self.programmable_constraint.validate_input_data(data)
 
     def _get_updated_metadata(self, metadata):
@@ -180,13 +203,41 @@ class ProgrammableConstraintHarness(BaseConstraint):
         return self.programmable_constraint.get_updated_metadata(metadata)
 
     def _fit(self, data, metadata):
+        if self._is_programmable_single_table_constraint:
+            data = data[self._table_name]
+
         self.programmable_constraint.fit(data, metadata)
 
     def _transform(self, data):
-        return self.programmable_constraint.transform(data)
+        if self._is_programmable_single_table_constraint:
+            data = data[self._table_name]
+
+        transformed = self.programmable_constraint.transform(data)
+
+        if self._is_programmable_single_table_constraint:
+            return {self._table_name: transformed}
+
+        return transformed
 
     def _reverse_transform(self, data):
-        return self.programmable_constraint.reverse_transform(data)
+        if self._is_programmable_single_table_constraint:
+            data = data[self._table_name]
+
+        reverse_transformed = self.programmable_constraint.reverse_transform(data)
+
+        if self._is_programmable_single_table_constraint:
+            return {self._table_name: reverse_transformed}
+
+        return reverse_transformed
 
     def _is_valid(self, data, metadata):
-        return self.programmable_constraint.is_valid(data)
+        if self._is_programmable_single_table_constraint:
+            table_name = self._get_single_table_name(metadata)
+            data = data[table_name]
+
+        is_valid = self.programmable_constraint.is_valid(data)
+
+        if self._is_programmable_single_table_constraint:
+            return {table_name: is_valid}
+
+        return is_valid

{sdv-1.37.1.dev0 → sdv-1.37.2.dev0/sdv.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sdv
-Version: 1.37.1.dev0
+Version: 1.37.2.dev0
 Summary: Generate synthetic data for single table, multi table and sequential data
 Author-email: "DataCebo, Inc." <info@sdv.dev>
 License-Expression: BUSL-1.1