Function columnar_array

tests/unit/test_data_handling.py:20–27

(component_type, n_components, attributes=None, batch_size_tuple=())

def columnar_array(component_type, n_components, attributes=None, batch_size_tuple=()):
    component_dtype = power_grid_meta_data[DT.sym_output][component_type].dtype
    required_attributes = (
        set(component_dtype.names) & set(attributes) if attributes is not None else component_dtype.names
    )
    return {
        attr: np.empty((n_components, *batch_size_tuple), dtype=component_dtype[attr]) for attr in required_attributes
    }


def row_array(component_type, n_components, batch_size_tuple=()):