diff --git a/rb/AN_RB_FPA.ipynb b/rb/AN_RB_FPA.ipynb index 0be3cecde19ea2fda84ee81ef391189a57bfdafc..3ff3d98fcb428eeeab942e2ead205eda70e820dd 100644 --- a/rb/AN_RB_FPA.ipynb +++ b/rb/AN_RB_FPA.ipynb @@ -1105,8 +1105,8 @@ }, "outputs": [], "source": [ - "def collection_was_stopped(u_diode_rb_df):\n", - " return u_diode_rb_df.index[-1] <= 0\n", + "def collection_was_stopped(u_diode_rb_df: pd.DataFrame):\n", + " return u_diode_rb_df.empty or u_diode_rb_df.index[-1] <= 0\n", "\n", "magnets_expected = MappingMetadata.get_magnets_for_circuit_names(circuit_type, circuit_name)\n", "magnets_found = {df.columns[0].partition(':')[0] for df in u_diode_rb_dfs if not collection_was_stopped(df)}\n",