Skip to content

Commit f63ec91

Browse files
authored
Update the warning that's displayed when using HMA on complex schemas (#2333)
1 parent fc3582d commit f63ec91

File tree

4 files changed

+9
-7
lines changed

4 files changed

+9
-7
lines changed

sdv/multi_table/hma.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -249,9 +249,9 @@ def _print_estimate_warning(self):
249249
+ '\n'
250250
)
251251
self._print(
252-
"We recommend simplifying your metadata schema using 'sdv.utils.poc.simplify_sch"
253-
"ema'.\nIf this is not possible, contact us at [email protected] for enterprise solut"
254-
'ions.\n'
252+
'We recommend simplifying your metadata schema using '
253+
"'sdv.utils.poc.simplify_schema'.\nIf this is not possible, please visit "
254+
'datacebo.com and reach out to us for enterprise solutions.\n'
255255
)
256256

257257
def preprocess(self, data):

tests/integration/multi_table/test_hma.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -722,7 +722,7 @@ def test_warning_message_too_many_cols(self, capsys):
722722
key_phrases = [
723723
r'PerformanceAlert:',
724724
r'large number of columns.',
725-
r'contact us at [email protected] for enterprise solutions.',
725+
r'please visit datacebo.com and reach out to us for enterprise solutions.',
726726
]
727727

728728
# Run

tests/integration/utils/test_poc.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -76,8 +76,10 @@ def test_simplify_schema(capsys):
7676
r'matches\s*39\s*412\s*'
7777
r'players\s*5\s*378\s*'
7878
r'teams\s*1\s*173004\s*'
79-
r"We recommend simplifying your metadata schema using 'sdv.utils.poc.simplify_schema'\.\s*"
80-
r'If this is not possible, contact us at [email protected] for enterprise solutions\.'
79+
r'We recommend simplifying your metadata schema using '
80+
r"'sdv.utils.poc.simplify_schema'\.\s*"
81+
r'If this is not possible, please visit '
82+
r'datacebo.com and reach out to us for enterprise solutions\.'
8183
)
8284
expected_message_after = re.compile(
8385
r'Success! The schema has been simplified\.\s+'

tests/unit/multi_table/test_hma.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ def test__print_estimate_warning(self, get_distributions_mock, estimate_mock, ca
109109
key_phrases = [
110110
r'PerformanceAlert:',
111111
r'large number of columns.',
112-
r'contact us at [email protected] for enterprise solutions.',
112+
r'please visit datacebo.com and reach out to us for enterprise solutions.',
113113
]
114114

115115
# Run

0 commit comments

Comments
 (0)