fix: nonavg don't use redgrp object

HERA-Team · Aug 5, 2024 · 64ccf0b · 64ccf0b
1 parent 31db705
commit 64ccf0b
Showing 1 changed file with 6 additions and 68 deletions.
diff --git a/hera_notebook_templates/notebooks/lststack.ipynb b/hera_notebook_templates/notebooks/lststack.ipynb
@@ -842,7 +842,9 @@
    "outputs": [],
    "source": [
     "auto_stats = [\n",
-    "    lstmet.LSTBinStats.from_reduced_data(rdc=rdc, antpairs=stackconf.autopairs, pols=stackconf.pols, reds=reds_with_pols) for rdc in autos_lstavg\n",
+    "    lstmet.LSTBinStats.from_reduced_data(\n",
+    "        rdc=rdc, antpairs=stackconf.autopairs, pols=stackconf.pols, reds=reds_with_pols if data_is_redundantly_averaged else None\n",
+    "    ) for rdc in autos_lstavg\n",
     "]"
    ]
   },
@@ -1131,72 +1133,6 @@
     "    original_data_mean = [lstavg['data'].copy() for lstavg in cross_lstavg]"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "13f99665-3f4b-4e13-b306-6836d64f89b1",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# from copy import deepcopy"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "d877cd3b-0a15-46da-bbd8-f89103abf1ce",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "# FILLED_CACHE = deepcopy(_INPAINT_CACHE_)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "c0fead3f-be97-4ae1-9bfc-d0315f323a98",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# from hera_cal.lst_stack import LSTStack"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "96cbe153-186e-42ad-82c6-4b82d63da2c3",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "# smallstack = LSTStack(stack.select(bls=stack.antpairs[:30], inplace=False))"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "2a9e0899-6786-40ea-a44e-51b693ed6b94",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# %load_ext line_profiler"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "293728f0-2894-4404-8bf9-4739a0b0441b",
-   "metadata": {
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "# %lprun -f avg.average_and_inpaint_simultaneously_single_bl avg.average_and_inpaint_simultaneously(smallstack, auto_stacks[0], inpaint_bands = inpaint_bands, return_models = make_plots, cache = {}, filter_properties = {\"min_dly\": inpaint_mindelay,  \"horizon\": inpaint_horizon, \"standoff\": inpaint_standoff},eigenval_cutoff=[inpaint_eigencutoff], max_gap_factor=inpaint_max_gap_factor,max_convolved_flag_frac=inpaint_max_convolved_flag_frac,use_unbiased_estimator=inpaint_use_unbiased_estimator,sample_cov_fraction=inpaint_sample_cov_fraction)\n"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -1560,7 +1496,9 @@
    "outputs": [],
    "source": [
     "cross_stats = [\n",
-    "    lstmet.LSTBinStats.from_reduced_data(rdc=rdc, antpairs=cross_stacks[0].antpairs, pols=stackconf.pols, reds=reds_with_pols) for rdc in cross_lstavg\n",
+    "    lstmet.LSTBinStats.from_reduced_data(\n",
+    "        rdc=rdc, antpairs=cross_stacks[0].antpairs, pols=stackconf.pols, reds=reds_with_pols if data_is_redundantly_averaged else None\n",
+    "    ) for rdc in cross_lstavg\n",
     "]"
    ]
   },