correct return val

HERA-Team · Mar 3, 2025 · b1e074c · b1e074c
1 parent 54a8c6f
commit b1e074c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hera_notebook_templates/notebooks/lststack.ipynb b/hera_notebook_templates/notebooks/lststack.ipynb
@@ -1264,7 +1264,7 @@
     "\n",
     "    for i, stack in enumerate(auto_stacks):\n",
     "        \n",
-    "        _avg, dpss_models = avg.average_and_inpaint_simultaneously(\n",
+    "        _avg, dpss_models, inp_flags = avg.average_and_inpaint_simultaneously(\n",
     "            stack,\n",
     "            stack,\n",
     "            inpaint_bands = inpaint_bands,\n",
@@ -1686,7 +1686,7 @@
     "    inpaint_dpss_models = []\n",
     "\n",
     "    for i, (stack, auto_stack) in enumerate(zip(cross_stacks, auto_stacks)):\n",
-    "        _avg, dpss_models = avg.average_and_inpaint_simultaneously(\n",
+    "        _avg, dpss_models, inp_flags = avg.average_and_inpaint_simultaneously(\n",
     "            stack,\n",
     "            auto_stack,\n",
     "            inpaint_bands = inpaint_bands,\n",