Skip to content

Commit 7245b35

Browse files
committed
Run serially when n_process==1
1 parent 619b7f1 commit 7245b35

1 file changed

Lines changed: 17 additions & 8 deletions

File tree

src/aspire/apple/apple.py

Lines changed: 17 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -165,16 +165,25 @@ def process_folder(self, folder, create_jpg=False):
165165
logger.info(f"launching {self.n_processes} processes")
166166

167167
pbar = tqdm(total=len(filenames))
168-
with futures.ProcessPoolExecutor(self.n_processes) as executor:
169-
to_do = []
170-
for filename in filenames:
171-
future = executor.submit(self.process_micrograph, filename, create_jpg)
172-
to_do.append(future)
173168

174-
for future in futures.as_completed(to_do):
175-
# Retrieve (None) result, since this operation re-raises Exceptions, if any.
176-
_ = future.result()
169+
if self.n_processes == 1:
170+
for filename in filenames:
171+
self.process_micrograph(filename, create_jpg)
177172
pbar.update(1)
173+
else:
174+
with futures.ProcessPoolExecutor(self.n_processes) as executor:
175+
to_do = []
176+
for filename in filenames:
177+
future = executor.submit(
178+
self.process_micrograph, filename, create_jpg
179+
)
180+
to_do.append(future)
181+
182+
for future in futures.as_completed(to_do):
183+
# Retrieve (None) result, since this operation re-raises Exceptions, if any.
184+
_ = future.result()
185+
pbar.update(1)
186+
178187
pbar.close()
179188

180189
def process_micrograph_centers(

0 commit comments

Comments
 (0)