pingu_98/cara
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added type hint for axes (matplotlib update)

Browse source
This commit is contained in:
lrdossan 2024-07-11 17:34:35 +02:00
parent 56550c4406
commit d4455aaa27
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
caimira/apps/calculator/report_generator.py
View file

@ -279,11 +279,14 @@ def uncertainties_plot(infection_probability: models._VectorisedFloat,
lower_percentiles: models._VectorisedFloat,
upper_percentiles: models._VectorisedFloat):
fig, axs = plt.subplots(2, 3,
fig, axes = plt.subplots(2, 3,
gridspec_kw={'width_ratios': [5, 0.5] + [1],
'height_ratios': [3, 1], 'wspace': 0},
sharey='row',
sharex='col')
# Type hint for axs
axs: np.ndarray = np.array(axes)
for y, x in [(0, 1)] + [(1, i + 1) for i in range(2)]:
axs[y, x].axis('off')