From 5c692000c59eac1133de026178e4b989162963f6 Mon Sep 17 00:00:00 2001 From: Felipe Date: Wed, 8 Feb 2023 10:55:41 -0800 Subject: [PATCH] Fix keys --- sdv/metadata/single_table.py | 11 ++--------- 1 file changed, 2 insertions(+), 9 deletions(-) diff --git a/sdv/metadata/single_table.py b/sdv/metadata/single_table.py index d172933d0..bae3376bd 100644 --- a/sdv/metadata/single_table.py +++ b/sdv/metadata/single_table.py @@ -214,11 +214,7 @@ def to_dict(self): """Return a python ``dict`` representation of the ``SingleTableMetadata``.""" metadata = {} for key in self._KEYS: - if key in ('columns', 'primary_key', 'alternate_keys', 'sequence_key', 'sequence_index'): - value = getattr(self, f'{key}') - else: - value = \ - getattr(self, f'_{key}') if key != 'METADATA_SPEC_VERSION' else self._version + value = getattr(self, f'{key}') if key != 'METADATA_SPEC_VERSION' else self._version if value: metadata[key] = value @@ -484,10 +480,7 @@ def _load_from_dict(cls, metadata): for key in instance._KEYS: value = deepcopy(metadata.get(key)) if value: - if key in ('columns', 'primary_key', 'alternate_keys', 'sequence_key', 'sequence_index'): - setattr(instance, f'{key}', value) - else: - setattr(instance, f'_{key}', value) + setattr(instance, f'{key}', value) return instance