diff options
Diffstat (limited to 'src/argaze/AreaOfInterest/AOIFeatures.py')
-rw-r--r-- | src/argaze/AreaOfInterest/AOIFeatures.py | 15 |
1 files changed, 3 insertions, 12 deletions
diff --git a/src/argaze/AreaOfInterest/AOIFeatures.py b/src/argaze/AreaOfInterest/AOIFeatures.py index e452f05..ff33a29 100644 --- a/src/argaze/AreaOfInterest/AOIFeatures.py +++ b/src/argaze/AreaOfInterest/AOIFeatures.py @@ -268,7 +268,7 @@ class AreaOfInterest(numpy.ndarray): AOISceneType = TypeVar('AOIScene', bound="AOIScene") # Type definition for type annotation convenience -class AOIScene(): +class AOIScene(DataFeatures.PipelineStepObject): """Define AOI scene as a dictionary of AOI.""" def __init__(self, dimension: int, areas: dict = None): @@ -276,6 +276,8 @@ class AOIScene(): assert(dimension > 0) + super().__init__() + self.__dimension = dimension self.__areas = {} @@ -395,17 +397,6 @@ class AOIScene(): return str(self.__areas) - def __str__(self) -> str: - """String display""" - - output = '' - - for name, area in self.__areas.items(): - - output += f'\n\t{name}:\n{area}' - - return output - def __add__(self, add_vector) -> AOISceneType: """Add vector to scene.""" |