Progress

Author: daquinteroflex

tidy3d/plugins/smatrix/data/terminal.py

@@ -78,7 +78,7 @@ class TerminalComponentModelerData(Tidy3dBaseModel):
     @cached_property
     def smatrix(self) -> MicrowaveSMatrixData:
         "Stores the computed S-matrix and reference impedances for the terminal ports"
-        from tidy3d.plugins.smatrix.local_run import construct_smatrix
+        from tidy3d.plugins.smatrix.run.local_run import construct_smatrix
 
         terminal_port_data = construct_smatrix(simulation=self.simulation)
         smatrix_data = MicrowaveSMatrixData(data=terminal_port_data)
@@ -141,7 +141,7 @@ def get_antenna_metrics_data(
             Container with antenna parameters including directivity, gain, and radiation efficiency,
             computed from the superposition of fields from all excited ports.
         """
-        from tidy3d.plugins.smatrix.local_run import compute_power_wave_amplitudes_at_each_port
+        from tidy3d.plugins.smatrix.run.local_run import compute_power_wave_amplitudes_at_each_port
 
         # Use the first port as default if none specified
         if port_amplitudes is None:
@@ -210,18 +210,21 @@ def get_antenna_metrics_data(
     @cached_property
     def port_reference_impedances(self) -> PortDataArray:
         """The reference impedance used at each port for definining power wave amplitudes."""
-        from tidy3d.plugins.smatrix.local_run import port_reference_impedances
+        from tidy3d.plugins.smatrix.run.local_run import port_reference_impedances
 
         return port_reference_impedances(self.simulation)
 
     def compute_power_wave_amplitudes_at_each_port(
         self, port_reference_impedances: PortDataArray, sim_data: SimulationData
     ) -> tuple[PortDataArray, PortDataArray]:
-        from tidy3d.plugins.smatrix.local_run import compute_power_wave_amplitudes_at_each_port
+        from tidy3d.plugins.smatrix.run.local_run import compute_power_wave_amplitudes_at_each_port
 
         data = compute_power_wave_amplitudes_at_each_port(
             simulation=self.simulation,
             port_reference_impedances=port_reference_impedances,
             sim_data=sim_data,
         )
         return data
+
+    # Mirror Utils
+    # So they can be reused elsewhere without a class reimport