Skip to content

Commit 50969b7

Browse files
committed
Bug fix for scale and shift in Layout and TextBlock
1 parent 33995d5 commit 50969b7

File tree

2 files changed

+12
-8
lines changed

2 files changed

+12
-8
lines changed

src/layoutparser/elements.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1234,16 +1234,16 @@ def is_in(self, other, **kwargs):
12341234
return self.block.is_in(other, **kwargs)
12351235

12361236
@mixin_textblock_meta
1237-
def shift(self, **kwargs):
1238-
return self.block.shift(**kwargs)
1237+
def shift(self, shift_distance):
1238+
return self.block.shift(shift_distance)
12391239

12401240
@mixin_textblock_meta
12411241
def pad(self, **kwargs):
12421242
return self.block.pad(**kwargs)
12431243

12441244
@mixin_textblock_meta
1245-
def scale(self, **kwargs):
1246-
return self.block.scale(**kwargs)
1245+
def scale(self, scale_factor):
1246+
return self.block.scale(scale_factor)
12471247

12481248
def crop_image(self, image):
12491249
return self.block.crop_image(image)
@@ -1301,16 +1301,16 @@ def filter_by(self, other):
13011301
return self.__class__([ele for ele in self if ele.is_in(other)])
13021302

13031303
@functools.wraps(BaseCoordElement.shift)
1304-
def shift(self, **kwargs):
1305-
return self.__class__([ele.shift(**kwargs) for ele in self])
1304+
def shift(self, shift_distance):
1305+
return self.__class__([ele.shift(shift_distance) for ele in self])
13061306

13071307
@functools.wraps(BaseCoordElement.pad)
13081308
def pad(self, **kwargs):
13091309
return self.__class__([ele.pad(**kwargs) for ele in self])
13101310

13111311
@functools.wraps(BaseCoordElement.scale)
1312-
def scale(self, **kwargs):
1313-
return self.__class__([ele.scale(**kwargs) for ele in self])
1312+
def scale(self, scale_factor):
1313+
return self.__class__([ele.scale(scale_factor) for ele in self])
13141314

13151315
def crop_image(self, image):
13161316
return [ele.crop_image(image) for ele in self]

tests/test_elements.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -162,6 +162,10 @@ def test_layout():
162162
l.filter_by(t)
163163
l.is_in(r)
164164

165+
l.scale(4)
166+
l.shift(4)
167+
l.pad(left=2)
168+
165169
def test_df():
166170

167171
df = pd.DataFrame(

0 commit comments

Comments
 (0)