official-apps/SeismicHazardForecasting
10
0
Fork 1
Code Issues 2 Pull Requests Activity

Update src/seismic_hazard_forecasting.py

Browse Source
This commit is contained in:
ftong 2025-06-06 19:14:55 +02:00
parent 59cc14e4da
commit 6dae0b6f07
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/seismic_hazard_forecasting.py
View File

@ -125,7 +125,7 @@ verbose: {verbose}")
logger.debug(f"BASELINE CPU LOAD% {psutil.cpu_percent(interval=None, percpu=True)}")
dask.config.set(scheduler='processes')
# run magnitude distribution modeling if selected by user and no magnitude pdf file provided
if m_select and m_range[0] == None and m_pdf[0] == None:
@ -444,6 +444,7 @@ verbose: {verbose}")
use_pp = True
if use_pp: # use dask parallel computing
dask.config.set(scheduler='processes', num_workers = 16)
pbar = ProgressBar()
pbar.register()
# iter = range(0,len(distances))
@ -454,7 +455,7 @@ verbose: {verbose}")
imls = [dask.delayed(compute_IMT_exceedance)(rx_lat[i], rx_lon[i], distances[i].flatten(), fr, p, lambdas,
forecast_len, lambdas_perc, m_range, m_pdf, m_cdf, model,
log_level=logging.DEBUG, imt=imt, IMT_min=0.0, IMT_max=2.0, rx_label=i,
imt=imt, IMT_min=0.0, IMT_max=2.0, rx_label=i,
rtol=0.1, use_cython=False) for i in iter]
iml = dask.compute(*imls)
iml_grid_raw.append(list(iml))