diff --git a/tab2dict/tab_dict.py b/tab2dict/tab_dict.py index 86a62db..52e3cbb 100644 --- a/tab2dict/tab_dict.py +++ b/tab2dict/tab_dict.py @@ -19,8 +19,6 @@ def __init__( super().__init__() self.tdict_type = tdict_type self.key_cols = key_cols - if tdict_type == "Relation": - self.key_cols = key_cols[:-1] self._data = tdict_data def __len__(self): @@ -57,6 +55,8 @@ def from_dataframe( df=df, value_column_name=value_column_name, ) + if tdict_type == "Relation": + del key_cols[-1] return cls(tdict_type=tdict_type, key_cols=key_cols, tdict_data=tdict_data) @classmethod