PyCS/tests/client/test_models/simple_model/model.py

from pycs.interfaces.Pipeline import Pipeline
from pycs.interfaces.MediaFile import MediaFile
from pycs.interfaces.MediaStorage import MediaStorage

class Model(Pipeline):

    def __init__(self, root_folder: str, configuration: dict):
        super().__init__(root_folder, configuration)


    def close(self):
        print("Closing")

    def execute(self, storage: MediaStorage, file: MediaFile):
        file.add_bounding_box(0, 0, 1, 1/3)

    def fit(self, storage: MediaStorage):
        print("fitting model")