Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_register(self):
"""
Test register.
"""
dtype = "int"
def add1(t, x):
return x + 1
def flip_sub_width(t, x):
return x - t.width
T.Transform.register_type(dtype, add1)
T.HFlipTransform.register_type(dtype, flip_sub_width)
transforms = T.TransformList(
[
T.ScaleTransform(0, 0, 0, 0, 0),
T.CropTransform(0, 0, 0, 0),
T.HFlipTransform(3),
]
)
self.assertEqual(transforms.apply_int(3), 2)
with self.assertRaises(AssertionError):
T.HFlipTransform.register_type(dtype, lambda x: 1)
def __radd__(self, other: "TransformList") -> "TransformList":
"""
Args:
other (TransformList): transformation to add.
Returns:
TransformList: list of transforms.
"""
others = (
other.transforms if isinstance(other, TransformList) else [other]
)
return TransformList(others + self.transforms)
def __add__(self, other: "TransformList") -> "TransformList":
"""
Args:
other (TransformList): transformation to add.
Returns:
TransformList: list of transforms.
"""
others = (
other.transforms if isinstance(other, TransformList) else [other]
)
return TransformList(self.transforms + others)
TransformList: contain the transforms that's used.
"""
for g in transform_gens:
assert isinstance(g, TransformGen), g
check_dtype(img)
tfms = []
for g in transform_gens:
tfm = g.get_transform(img)
assert isinstance(
tfm, Transform
), "TransformGen {} must return an instance of Transform! Got {} instead".format(g, tfm)
img = tfm.apply_image(img)
tfms.append(tfm)
return img, TransformList(tfms)
def __radd__(self, other: "TransformList") -> "TransformList":
"""
Args:
other (TransformList): transformation to add.
Returns:
TransformList: list of transforms.
"""
others = (
other.transforms if isinstance(other, TransformList) else [other]
)
return TransformList(others + self.transforms)