Skip to content

Commit 41bfb0b

Browse files
authored
Fix print for GaussianNormalizer (#905)
1 parent ab520e8 commit 41bfb0b

File tree

2 files changed

+13
-1
lines changed

2 files changed

+13
-1
lines changed

rdt/transformers/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -343,7 +343,7 @@ def __repr__(self):
343343
instanced = {
344344
key: getattr(self, key)
345345
for key in keys
346-
if key != 'model_missing_values' # Remove after deprecation
346+
if key != 'model_missing_values' and hasattr(self, key) # Remove after deprecation
347347
}
348348

349349
if defaults == instanced:

tests/unit/transformers/test_numerical.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1176,6 +1176,18 @@ def test__reverse_transform_missing_value_generation(self):
11761176
# Assert
11771177
np.testing.assert_allclose(transformed_data, expected, rtol=1e-3)
11781178

1179+
def test_print(self, capsys):
1180+
"""Test the class can be printed. GH#883"""
1181+
# Setup
1182+
transformer = GaussianNormalizer()
1183+
1184+
# Run
1185+
print(transformer) # noqa: T201 `print` found
1186+
1187+
# Assert
1188+
captured = capsys.readouterr()
1189+
assert captured.out == 'GaussianNormalizer()\n'
1190+
11791191

11801192
class TestClusterBasedNormalizer(TestCase):
11811193
def test__get_current_random_seed_random_states_is_none(self):

0 commit comments

Comments
 (0)