fix EModelCreation

Change-Id: I5cbd4d3b8d315dca6cb66976ab224e2eff365b0f
BlueBrain · Aug 18, 2023 · fb4e637 · fb4e637
1 parent 03f4589
commit fb4e637
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/bluepyemodel/tasks/emodel_creation/optimisation.py b/bluepyemodel/tasks/emodel_creation/optimisation.py
@@ -19,6 +19,7 @@
 import glob
 import multiprocessing
 from pathlib import Path
+import logging
 
 import luigi
 
@@ -37,6 +38,7 @@
 from bluepyemodel.tools.utils import get_legacy_checkpoint_path
 
 # pylint: disable=W0235,W0621,W0404,W0611,W0703,E1128
+logger = logging.getLogger(__name__)
 
 
 def _reformat_ttype(ttype):

diff --git a/bluepyemodel/tasks/luigi_tools.py b/bluepyemodel/tasks/luigi_tools.py
@@ -78,7 +78,9 @@ def inner(self):
             """Inner decorator function"""
             if EmodelAPIConfig().api == "nexus":
                 self.access_point.check_mettypes()
-            func(self)
+            # do this instead of just func(self) because of the yield in EModelCreation
+            for x in func(self):
+                yield x
 
         return inner
 

diff --git a/bluepyemodel/version.py b/bluepyemodel/version.py
@@ -1,4 +1,4 @@
 """ Package version. """
 
-VERSION = "0.0.16.dev2"
+VERSION = "0.0.16.dev3"
 __version__ = VERSION