diff --git a/apps/Moehlis-Faisst-Eckhart.py b/apps/Moehlis-Faisst-Eckhart.py index e859290..3184c6e 100755 --- a/apps/Moehlis-Faisst-Eckhart.py +++ b/apps/Moehlis-Faisst-Eckhart.py @@ -210,6 +210,8 @@ def solve(u, dt, n): # produce and visualize data # Moehlis-Faisst-Eckhart model #========================================== + if not os.path.exists('MFE'): + os.mkdir('MFE') dt = 1.e-2 mfe = MoehlisFaisstEckhart(Re = 800., Lx = 4.*np.pi, Lz = 2.*np.pi) diff --git a/fds/slurm.py b/fds/slurm.py index 17711b3..e5a0c35 100644 --- a/fds/slurm.py +++ b/fds/slurm.py @@ -22,9 +22,9 @@ class grab_from_SLURM_NODELIST: ''' def __init__(self, num_nodes, lock_and_dict, exclude_this_node=False): if lock_and_dict is None: - self.lock = None - else: - self.lock, self.dict = lock_and_dict + self.lock = None + else: + self.lock, self.dict = lock_and_dict self.grab(num_nodes, exclude_this_node) def grab(self, num_nodes, exclude_this_node=False):