Adjusted tut1 to waves realization notation

sandialabs · Dec 8, 2023 · d0d6a84 · d0d6a84
1 parent c0e6c78
commit d0d6a84
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/examples/tutorial_1_WaveBot.ipynb b/examples/tutorial_1_WaveBot.ipynb
@@ -468,7 +468,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "p_flows = wot.utilities.calculate_power_flows(wec, pto, results, waves, intrinsic_impedance)\n",
+    "p_flows = wot.utilities.calculate_power_flows(wec, pto, results[0], waves.sel(realization=0), intrinsic_impedance)\n",
     "wot.utilities.plot_power_flow(p_flows)"
    ]
   },
@@ -677,7 +677,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "p_flows_2 = wot.utilities.calculate_power_flows(wec_2, pto_2, results_2, waves, intrinsic_impedance)\n",
+    "p_flows_2 = wot.utilities.calculate_power_flows(wec_2, pto_2, results_2[0], waves.sel(realization = 0), intrinsic_impedance)\n",
     "wot.utilities.plot_power_flow(p_flows_2)"
    ]
   },
@@ -694,7 +694,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "p_flows_2_nocon = wot.utilities.calculate_power_flows(wec_2_nocon, pto_2, results_2_nocon, waves, intrinsic_impedance)\n",
+    "p_flows_2_nocon = wot.utilities.calculate_power_flows(wec_2_nocon, pto_2, results_2_nocon[0], waves.sel(realization = 0), intrinsic_impedance)\n",
     "wot.utilities.plot_power_flow(p_flows_2_nocon)"
    ]
   },

diff --git a/wecopttool/core.py b/wecopttool/core.py
@@ -2270,7 +2270,8 @@ def hydrodynamic_impedance(hydro_data: Dataset) -> Dataset:
         + hydro_data['added_mass'])*1j*hydro_data['omega'] \
             + hydro_data['radiation_damping'] + hydro_data['friction'] \
                 + hydro_data['hydrostatic_stiffness']/1j/hydro_data['omega']
-    return Zi.transpose('omega', 'radiating_dof', 'influenced_dof')
+    return intrinsic_impedance.transpose('omega', 'radiating_dof', 'influenced_dof')
+
 
 
 def atleast_2d(array: ArrayLike) -> ndarray: