class DummyData:
    def __init__(self, shape):
        self.shape = shape

class CustomShapeData:
    def __init__(self, custom_shape):
        self.custom_shape_attr = custom_shape

class CustomClass:
        pass