Think it'll work!

Author: daquinteroflex

tidy3d/plugins/smatrix/data/data.py

@@ -105,6 +105,8 @@ 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.run import compute_power_wave_amplitudes_at_each_port
+
         # Use the first port as default if none specified
         if port_amplitudes is None:
             port_amplitudes = {self.simulation.ports[0].name: None}
@@ -132,8 +134,10 @@ def get_antenna_metrics_data(
             sim_data_port = self.simulation.batch_data[self.simulation._task_name(port=port)]
             radiation_data = sim_data_port[rad_mon.name]
 
-            a, b = self.compute_power_wave_amplitudes_at_each_port(
-                self.port_reference_impedances, sim_data_port
+            a, b = compute_power_wave_amplitudes_at_each_port(
+                simulation=self.simulation,
+                port_reference_impedances=self.simulation.port_reference_impedances,
+                sim_data=sim_data_port,
             )
             # Select a possible subset of frequencies
             a = a.sel(f=f)
@@ -172,7 +176,7 @@ def port_reference_impedances(self) -> PortDataArray:
         """The reference impedance used at each port for definining power wave amplitudes."""
         from tidy3d.plugins.smatrix.run import _port_reference_impedances
 
-        return _port_reference_impedances(self.simulation.batch_data)
+        return _port_reference_impedances(self.simulation)
 
     def compute_power_wave_amplitudes_at_each_port(
         self, port_reference_impedances: PortDataArray, sim_data: SimulationData