Method __init__

climart/utils/callbacks.py:42–51

(self,
                 variables: List[str],
                 sizes: Union[int, Sequence[int]]
                 )

class PredictionPostProcessCallback:
    def __init__(self,
                 variables: List[str],
                 sizes: Union[int, Sequence[int]]
                 ):
        self.variable_to_channel = dict()
        cur = 0
        sizes = [sizes for _ in range(len(variables))] if isinstance(sizes, int) else sizes
        for var, size in zip(variables, sizes):
            self.variable_to_channel[var] = {'start': cur, 'end': cur + size}
            cur += size

    def split_vector_by_variable(self,
                                 vector: Union[np.ndarray, torch.Tensor]