From 1b7e9f76464bc4729f3f873345cf5b16c178f3dd Mon Sep 17 00:00:00 2001 From: ftong Date: Thu, 21 Aug 2025 12:11:02 +0200 Subject: [PATCH] Update src/seismic_hazard_forecasting.py --- src/seismic_hazard_forecasting.py | 20 ++++++++++++++------ 1 file changed, 14 insertions(+), 6 deletions(-) diff --git a/src/seismic_hazard_forecasting.py b/src/seismic_hazard_forecasting.py index 4ebc6ba..f8c8afd 100644 --- a/src/seismic_hazard_forecasting.py +++ b/src/seismic_hazard_forecasting.py @@ -1,5 +1,9 @@ # -*- coding: utf-8 -*- - +def my_trivial_task(x): + """ + A function that does a trivial task + """ + return x * x def main(catalog_file, mc_file, pdf_file, m_file, m_select, mag_label, mc, m_max, m_kde_method, xy_select, grid_dim, xy_win_method, rate_select, time_win_duration, @@ -454,7 +458,7 @@ verbose: {verbose}") start = timer() - use_pp = False + use_pp = True if use_pp: # use dask parallel computing pbar = ProgressBar() @@ -465,13 +469,17 @@ verbose: {verbose}") for imt in products: logger.info(f"Estimating {imt}") - 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, - rtol=0.1, use_cython=True) for i in iter] + #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, + # rtol=0.1, use_cython=True) for i in iter] + imls = [dask.delayed(my_trivial_task)(i) for i in iter] iml = dask.compute(*imls) iml_grid_raw.append(list(iml)) + + + else: iml_grid_raw = [] iter = indices