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):
        print("executing model")

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