Source code for powersimdata.network.usa_tamu.model

import os

from powersimdata.input.converter.csv_to_grid import FromCSV
from powersimdata.network.constants.carrier.storage import get_storage
from powersimdata.network.helpers import check_and_format_interconnect
from powersimdata.network.model import ModelImmutables


class TAMU(FromCSV):
    """TAMU network.

    :param str/iterable interconnect: interconnect name(s).
    """

    def __init__(self, interconnect):
        """Constructor."""
        super().__init__()

        self.grid_model = "usa_tamu"
        self.interconnect = check_and_format_interconnect(
            interconnect, model=self.grid_model
        )
        self.model_immutables = ModelImmutables(
            self.grid_model, interconnect=interconnect
        )
        self._set_data_loc(os.path.dirname(__file__))

    def build(self):
        """Build network"""
        self._build(self.interconnect, self.grid_model)
        self.storage.update(get_storage(self.grid_model))