From e42ad75fa4b0815db7d44bd7e081ef4c847509f5 Mon Sep 17 00:00:00 2001 From: Fischer Robert Date: Wed, 27 Sep 2023 13:57:14 +0200 Subject: [PATCH] fix identation --- training_functions.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/training_functions.py b/training_functions.py index 482ffe0..59ea617 100644 --- a/training_functions.py +++ b/training_functions.py @@ -373,21 +373,21 @@ class train_segmentation: fut = fut.result() fut = fut.compute() feat_stack = fut.data - try: - self.client.restart() + try: + self.client.restart() except: # if not len(self.client.cluster.workers)>1: - self.client = reboot_client(self.client, memory_limit=self.memlim, n_workers=self.n_workers) + self.client = reboot_client(self.client, memory_limit=self.memlim, n_workers=self.n_workers) # TODO client reboot if workers can't return if type(feat_stack_t_idp) is not np.ndarray: fut = self.client.scatter(feat_stack_t_idp) fut = fut.result() fut = fut.compute() feat_stack_t_idp = fut.data - try: + try: self.client.restart() # if not len(self.client.cluster.workers)>1: - except: + except: self.client = reboot_client(self.client, memory_limit=self.memlim, n_workers=self.n_workers) feat_stack = np.concatenate([feat_stack, feat_stack_t_idp], axis = 2)