Skip to content

function_configs

Bases: config

Source code in tinybig/config/function_configs.py 
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
class function_configs(config):
    def __init__(
        self,
        configs: dict | list = None,
        class_name: str = None,
        parameters: dict = None,
        name: str = 'function_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 not class_name.startswith('tinybig'):
                class_name = class_name.split('.')[-1]
                class_name = find_class_in_package(class_name=class_name)
            assert class_name.startswith('tinybig')
            configs = {
                'function_class': class_name,
                'function_parameters': 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=self.configs,
            device=self.device,
            class_name='function_class',
            parameter_name='function_parameters'
        )