fix new use in examples/demo_theory_functionality_oo.ipynb

LSSTDESC · Nov 16, 2023 · 5e8af97 · 5e8af97
1 parent 1b29e0b
commit 5e8af97
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/examples/demo_theory_functionality_oo.ipynb b/examples/demo_theory_functionality_oo.ipynb
@@ -132,8 +132,19 @@
     "\n",
     "gt = moo.eval_reduced_tangential_shear(r3d, z_cl, z_src)\n",
     "# Lensing quantities assuming sources follow a given redshift distribution.\n",
+    "\n",
+    "# Compute first beta\n",
+    "beta_kwargs = {\n",
+    "    \"z_cl\": z_cl,\n",
+    "    \"z_inf\": 10.0,\n",
+    "    \"cosmo\": cosmo,\n",
+    "    \"z_distrib_func\": z_distrib_func,\n",
+    "}\n",
+    "beta_s_mean = clmm.utils.compute_beta_s_mean_from_distribution(**beta_kwargs)\n",
+    "beta_s_square_mean = clmm.utils.compute_beta_s_square_mean_from_distribution(**beta_kwargs)\n",
+    "\n",
     "gt_z = moo.eval_reduced_tangential_shear(\n",
-    "    r3d, z_cl, z_distrib_func, z_src_info=\"distribution\", approx=\"order2\"\n",
+    "    r3d, z_cl, [beta_s_mean, beta_s_square_mean], z_src_info=\"beta\", approx=\"order2\"\n",
     ")\n",
     "\n",
     "mu = moo.eval_magnification(r3d, z_cl, z_src)\n",