Add add-on detection for constraints (#1412)

* add ability to find constraint addons

* fix error message

* fix test

sdv/_addons.py

@@ -21,7 +21,7 @@ def _find_addons(group, parent_globals, add_all=False):
         try:
             module = entry_point.load()
         except Exception:
-            msg = f'Failed to load "{entry_point.name}" from "{entry_point.module}".'
+            msg = f'Failed to load "{entry_point.name}" from "{entry_point.module_name}".'
             warnings.warn(msg)
             continue
 

sdv/constraints/__init__.py

@@ -1,5 +1,5 @@
 """SDV Constraints module."""
-
+from sdv._addons import _find_addons
 from sdv.constraints.base import Constraint
 from sdv.constraints.tabular import (
     FixedCombinations, FixedIncrements, Inequality, Negative, OneHotEncoding, Positive, Range,
@@ -19,3 +19,5 @@
     'OneHotEncoding',
     'Unique'
 ]
+
+_find_addons('sdv.constraints_modules', globals(), add_all=True)

tests/unit/test__addons.py

@@ -80,7 +80,7 @@ def entry_point_error():
 
     bad_entry_point = Mock()
     bad_entry_point.name = 'bad_entry_point'
-    bad_entry_point.module = 'bad_module'
+    bad_entry_point.module_name = 'bad_module'
     bad_entry_point.load.side_effect = entry_point_error
     entry_points_mock.return_value = [bad_entry_point]
     test_dict = {}