diff --git a/hera_notebook_templates/notebooks/lststack.ipynb b/hera_notebook_templates/notebooks/lststack.ipynb index d3e3505..e09768e 100644 --- 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", "]" ] },