tab_dict initialization bug fixed

SongminYu · Jan 22, 2024 · 291fc8b · 291fc8b
1 parent edcc367
commit 291fc8b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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