Commit 4bde2279 authored by Mukherjee, Debangshu's avatar Mukherjee, Debangshu
Browse files

Solved dask n_workers bug

parent d479f13a
......@@ -1595,6 +1595,12 @@ def strain_figure(exx, exy, eth, eyy, ROI, scale=0, scale_unit="nm", figsize=(22
be assigned too.
"""
# Convert NaN values to zero before plotting
exx[np.isnan(exx)] = 0
exy[np.isnan(exy)] = 0
eth[np.isnan(eth)] = 0
eyy[np.isnan(eyy)] = 0
def ROI_RdBu_map(valmap, roi, valrange):
plot_col = np.zeros((256, 3), dtype=np.float)
for ii in range(255):
......
......@@ -522,7 +522,7 @@ def generate4D_frms6(data_dir, bin_factor=2):
filesizes = np.zeros((tot_files, 4), dtype=int)
filenames = np.zeros(tot_files, dtype=object)
cluster = dd.LocalCluster(int(tot_files))
cluster = dd.LocalCluster(n_workers=int(1+tot_files))
client = dd.Client(cluster)
ii = 0
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment