Update src/seismic_hazard_forecasting.py

src/seismic_hazard_forecasting.py

@@ -73,7 +73,7 @@ def main(catalog_file, mc_file, pdf_file, m_file, m_select, mag_label, mc, m_max
     from math import ceil, floor, isnan
     import numpy as np
     import dask
-    # from dask.diagnostics import ProgressBar  # use Dask progress bar
+    from dask.diagnostics import ProgressBar  # use Dask progress bar
     import kalepy as kale
     import utm
     from skimage.transform import resize
@@ -90,7 +90,7 @@ def main(catalog_file, mc_file, pdf_file, m_file, m_select, mag_label, mc, m_max
     from matplotlib.contour import ContourSet
     import xml.etree.ElementTree as ET
     import json
-    from concurrent.futures import ThreadPoolExecutor
+    import multiprocessing as mp
 
     logger = getDefaultLogger('igfash')
 
@@ -133,7 +133,7 @@ verbose: {verbose}")
     logger.debug(f"Rbeast version {version('rbeast')}")
     logger.debug(f"Dask version {version('dask')}")
 
-    dask.config.set(scheduler='threads', pool=ThreadPoolExecutor(16))
+    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:
@@ -479,6 +479,7 @@ verbose: {verbose}")
         use_pp = True
 
         if use_pp:         # use dask parallel computing
+            mp.set_start_method("fork", force=True)
             #pbar = ProgressBar()
             #pbar.register()
             # iter = range(0,len(distances))