@@ -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:
@@ -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