import yaml
from oop_figures.loader.loader import Loader


class YamlLoader(Loader):
    def __init__(self, file_name):
        super().__init__(file_name)

    def load(self):
        with open(self.file_name) as f:
            load_input = yaml.load(f)
            return load_input