Merge pull request #127 from MaxFBurg/master

Fixes bug: resolve str function name
sinzlab · May 6, 2021 · 749f13f · 749f13f
2 parents 0f3e2b7 + b4e7bd3
commit 749f13f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nnfabrik/templates/trained_model.py b/nnfabrik/templates/trained_model.py
@@ -4,7 +4,7 @@
 import warnings
 import datajoint as dj
 from datajoint.fetch import DataJointError
-from ..builder import get_all_parts, get_model, get_trainer
+from ..builder import get_all_parts, get_model, get_trainer, resolve_fn
 from ..utility.dj_helpers import make_hash
 from .utility import find_object
 
@@ -327,6 +327,7 @@ def make(self, key):
                         }
         """
         dataset_fn, dataset_config = (self.dataset_table & key).fn_config
+        dataset_fn = resolve_fn(dataset_fn, "datasets")
         data_info = dataset_fn(**dataset_config, return_data_info=True)
 
         fabrikant_name = self.user_table.get_current_user()