johannesschmude commited on
Commit
7adac5a
·
1 Parent(s): d4d385c

Bug fix -- incorrect color scale.

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -231,9 +231,8 @@ def run_inference(init_time_idx, plt_channel_idx, hours_ahead):
231
  sl_scale_factor=sl_scale_factors[plt_channel_idx],
232
  )
233
  )
234
- vmin = max(vmin, sample_data["ts"][plt_channel_idx, i].min())
235
- #vmax = min(vmax, np.quantile(sample_data["ts"][plt_channel_idx, i], 0.99))
236
- vmax = min(vmax, sample_data["ts"][plt_channel_idx, i].max())
237
 
238
  if plt_channel_str.startswith("aia"):
239
  cm_name = "sdo" + plt_channel_str
 
231
  sl_scale_factor=sl_scale_factors[plt_channel_idx],
232
  )
233
  )
234
+ vmin = max(vmin, input_image[i].min())
235
+ vmax = min(vmax, np.quantile(input_image[i], 0.99))
 
236
 
237
  if plt_channel_str.startswith("aia"):
238
  cm_name = "sdo" + plt_channel_str