from pynwb.device import Device
from rec_to_nwb.processing.nwb.components.electrode_group.fl_electrode_group import \
    FlElectrodeGroup
from rec_to_nwb.processing.tools.beartype.beartype import beartype


class FlElectrodeGroupBuilder:

    @staticmethod
    @beartype
    def build(metadata: dict, device: Device):
        return FlElectrodeGroup(
            name=str(metadata["id"]),
            description=metadata['description'],
            location=metadata['location'],
            device=device
        )

from pynwb.device import Device
from rec_to_nwb.processing.nwb.components.electrode_group.fl_electrode_group import \
    FlElectrodeGroup
from rec_to_nwb.processing.tools.beartype.beartype import beartype


class FlElectrodeGroupBuilder:

    @staticmethod
    @beartype
    def build(metadata: dict, device: Device):
        return FlElectrodeGroup(
            name=str(metadata["id"]),
            description=metadata['description'],
            location=metadata['location'],
            device=device
        )
