Skip to content

function_list_configs

Bases: config

Source code in tinybig/config/function_configs.py 
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
class function_list_configs(config):
    def __init__(
        self,
        configs: dict | list = None,
        class_name: list | str = None,
        parameters: list | dict = None,
        name: str = 'function_list_configs',
        device: str = 'cpu',
        *args, **kwargs
    ):
        if configs is not None:
            super().__init__(name=name, configs=configs, device=device, *args, **kwargs)
        elif class_name is not None:
            if isinstance(class_name, str):
                class_name = [class_name]
            if isinstance(parameters, dict):
                parameters = [parameters]
            assert len(class_name) == len(parameters)
            configs = [
                function_configs(
                    class_name=name,
                    parameters=parameter,
                ).get_configs()
                for name, parameter in zip(class_name, parameters)
            ]
            super().__init__(name=name, configs=configs, device=device, *args, **kwargs)
        else:
            super().__init__(name=name, configs=None, device=device, *args, **kwargs)

    def to_instantiation(self):
        if self.configs is None:
            raise ValueError('The function configs object cannot be none...')

        return [config.instantiation_from_configs(
            configs=config,
            device=self.device,
            class_name='function_class',
            parameter_name='function_parameters'
        ) for config in self.configs]