Skip to content

Commit 0105671

Browse files
committed
add dump/load
1 parent 32d3cb3 commit 0105671

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

physics_atv_visual_mapping/feature_key_list.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,4 +88,11 @@ def filter_metainfo(self, k):
8888
Return a FeatureKeyList containing fks from self that have metainfo=k
8989
"""
9090
idxs = [i for i,m in enumerate(self.metainfo) if m==k]
91-
return self[idxs]
91+
return self[idxs]
92+
93+
def dump(self):
94+
return [f"{label}, {meta}" for label, meta in zip(self.label, self.metainfo)]
95+
96+
def load(input):
97+
labels, metas = zip(*[s.split(', ') for s in input])
98+
return FeatureKeyList(label=list(labels), metainfo=list(metas))

0 commit comments

Comments
 (0)