浏览代码

the brain model know now its how yaml config

nico 8 年之前
父节点
当前提交
c2fca5b501
共有 2 个文件被更改,包括 3 次插入1 次删除
  1. 1 0
      core/ConfigurationManager/BrainLoader.py
  2. 2 1
      core/Models/Brain.py

+ 1 - 0
core/ConfigurationManager/BrainLoader.py

@@ -38,6 +38,7 @@ class BrainLoader(object):
         dict_brain = cls.get_yaml_config(file_path)
         # create a new brain
         brain = Brain()
+        brain.brain_yaml = dict_brain
         # create list of Synapse
         synapses = list()
         for synapses_dict in dict_brain:

+ 2 - 1
core/Models/Brain.py

@@ -1,5 +1,6 @@
 
 class Brain(object):
-    def __init__(self, synapses=None, brain_file=None):
+    def __init__(self, synapses=None, brain_file=None, brain_yaml=None):
         self.synapses = synapses
         self.brain_file = brain_file
+        self.brain_yaml = brain_yaml