Skip to content
Snippets Groups Projects
Commit 84267d82 authored by Bas Nijholt's avatar Bas Nijholt
Browse files

Fix get_results()

parent 4f7b9bab
No related branches found
No related tags found
No related merge requests found
...@@ -171,10 +171,11 @@ class Learner1D(object): ...@@ -171,10 +171,11 @@ class Learner1D(object):
def get_results(self): def get_results(self):
"""Work with distributed.client.Future objects.""" """Work with distributed.client.Future objects."""
for x, y in self.unfinished.items(): done = [(x, y.result()) for x, y in self.unfinished.items() if y.done()]
if y.done(): for x, y in done:
y = self.unfinished.pop(x).result() self.unfinished.pop(x)
self.add_point(x, y) if done:
self.add_data(*np.array(done).T)
def add_futures(self, xs, ys): def add_futures(self, xs, ys):
"""Add concurrent.futures to the self.unfinished dict.""" """Add concurrent.futures to the self.unfinished dict."""
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment