continuing to try to fix mypy ax array errors

AllenCell · Dec 2, 2024 · bf362d4 · bf362d4
1 parent 0755338
commit bf362d4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nuc_morph_analysis/analyses/volume/FigS10_workflow.py b/nuc_morph_analysis/analyses/volume/FigS10_workflow.py
@@ -102,7 +102,8 @@
 # S10 panel C step3
 df_track = df_full[df_full.track_id == volume_dip_example_track]
 fig,axlist = plt.subplots(2,1,figsize=(fw,fh),sharey=False)
-axlist = list(axlist)
+axlist = np.asarray(axlist) if type(axlist) != np.ndarray else axlist # for mypy
+assert type(ax) == np.ndarray # for mypy
 
 _ = plot_track_with_fit_line(df_track,axlist[0],
                                 ycol1='volume',