diff --git a/src/seismic_hazard_forecasting.py b/src/seismic_hazard_forecasting.py index 1b49c0d..f6ebca0 100644 --- a/src/seismic_hazard_forecasting.py +++ b/src/seismic_hazard_forecasting.py @@ -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))