"""Module for Multi FeedForward model"""
import torch

from .feed_forward import FeedForward


class MultiFeedForward(torch.nn.Module):
    """
    :param dict dff_dict: dictionary of FeedForward networks.
    """
    def __init__(self, dff_dict):
        """
        """
        super().__init__()

        if not isinstance(dff_dict, dict):
            raise TypeError

        for name, constructor_args in dff_dict.items():
            setattr(self, name, FeedForward(**constructor_args))