data array map split

Author: marcorudolphflex

tidy3d/_common/components/data/data_array.py

@@ -43,9 +43,6 @@
     from tidy3d._common.components.types.base import Axis, Bound
 
 
-DATA_ARRAY_MAP: dict[str, type[DataArray]] = {}
-DATA_ARRAY_TYPES: list[type[DataArray]] = []
-
 # maps the dimension names to their attributes
 DIM_ATTRS = {
     "x": {"units": MICROMETER, "long_name": "x position"},
@@ -862,3 +859,13 @@ def reflect(self, axis: Axis, center: float, reflection_only: bool = False) -> S
         coords_dict = dict(zip("xyz", coords))
 
         return SpatialDataArray(new_data, coords=coords_dict)
+
+
+DATA_ARRAY_TYPES: list[type[DataArray]] = [
+    FreqDataArray,
+    TriangleMeshDataArray,
+    TimeDataArray,
+    SpatialDataArray,
+    ScalarFieldDataArray,
+]
+DATA_ARRAY_MAP = {data_array.__name__: data_array for data_array in DATA_ARRAY_TYPES}

tidy3d/components/data/data_array.py

@@ -864,8 +864,6 @@ def _make_impedance_data_array(result: DataArray) -> ImpedanceResultType:
 
 DATA_ARRAY_TYPES.extend(
     [
-        SpatialDataArray,
-        ScalarFieldDataArray,
         ScalarFieldTimeDataArray,
         ScalarModeFieldDataArray,
         FluxDataArray,
@@ -879,11 +877,8 @@ def _make_impedance_data_array(result: DataArray) -> ImpedanceResultType:
         FieldProjectionKSpaceDataArray,
         DiffractionDataArray,
         FreqModeDataArray,
-        FreqDataArray,
-        TimeDataArray,
         FreqModeDataArray,
         FreqVoltageDataArray,
-        TriangleMeshDataArray,
         HeatDataArray,
         EMEScalarFieldDataArray,
         EMEScalarModeFieldDataArray,