Skip to content

Commit 6bbf053

Browse files
committed
Add shape property
1 parent 29189fa commit 6bbf053

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

sota_extractor2/data/table.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -146,6 +146,10 @@ def set_tags(self, tags):
146146
# todo: change gold_tags to tags to avoid confusion
147147
self.df.iloc[r,c].gold_tags = cell.strip()
148148

149+
@property
150+
def shape(self):
151+
return self.df.shape
152+
149153
@property
150154
def matrix(self):
151155
return self.df.applymap(lambda x: x.value)

0 commit comments

Comments
 (0)