Bugfix tutorial helper function.

NBI current drive was not being updated consistently with the power in exercise 1.

PiperOrigin-RevId: 744656956

Author: jcitrin

torax/tutorials/torax_tutorial_exercises.ipynb

@@ -67,15 +67,15 @@
         "With Newton-Raphson stepper and adaptive timestep (backtracking)\n",
         "\"\"\"\n",
         "\n",
-        "_NBI_MW_TO_A = 1/16e6 # rough estimate of NBI heating power to current drive\n",
+        "_NBI_W_TO_MA = 1/16e6 # rough estimate of NBI heating power to current drive\n",
         "W_to_Ne_ratio = 0\n",
         "\n",
         "# No NBI during rampup. Rampup all NBI power between 99-100 seconds\n",
         "nbi_times = np.array([0, 99, 100])\n",
         "nbi_powers = np.array([0, 0, 33e6])\n",
-        "nbi_cd = nbi_powers * _NBI_MW_TO_A\n",
+        "nbi_cd = nbi_powers * _NBI_W_TO_MA\n",
         "\n",
-        "# Gaussian perscription of \"NBI\" deposition profiles and fractional deposition\n",
+        "# Gaussian prescription of \"NBI\" deposition profiles and fractional deposition\n",
         "r_nbi = 0.25\n",
         "w_nbi = 0.25\n",
         "el_heat_fraction = 0.66\n",
@@ -697,6 +697,7 @@
         "  config['runtime_params']['profile_conditions']['Ip_tot'] = {0: 3, LH_transition_time: 12.5}\n",
         "  config['sources']['electron_cyclotron_source']['gaussian_ec_total_power'] = {0: 0, LH_transition_time-1: 0, LH_transition_time: 20.0e6}\n",
         "  config['sources']['generic_ion_el_heat_source']['Ptot'] = {0: 0, LH_transition_time-1: 0, LH_transition_time: 33.0e6}\n",
+        "  config['sources']['generic_current_source']['Iext'] = {0: 0, LH_transition_time-1: 0, LH_transition_time: 33.0e6 * _NBI_W_TO_MA}\n",
         "  config['pedestal']['Tiped'] = {0: 0.5, LH_transition_time: 0.5, LH_transition_time+5: 3.0}\n",
         "  config['pedestal']['Teped'] = {0: 0.5, LH_transition_time: 0.5, LH_transition_time+5: 3.0}\n",
         "  return config\n",
@@ -773,7 +774,7 @@
         "  if nbi_power is not None:\n",
         "    _validate_input_dict(nbi_power, \"nbi_power\", (0.0, 33e6), \"W\")\n",
         "    config['sources']['generic_ion_el_heat_source']['Ptot'] = nbi_power\n",
-        "    nbi_current = {time: value*_NBI_MW_TO_A for time, value in nbi_power.items()}\n",
+        "    nbi_current = {time: value*_NBI_W_TO_MA for time, value in nbi_power.items()}\n",
         "    config['sources']['generic_current_source']['Iext'] = nbi_current\n",
         "  if off_axis_ec_power is not None:\n",
         "    _validate_input_dict(off_axis_ec_power, \"off_axis_ec_power\", (0.0, 40e6), \"W\")\n",