import tensorflow as tf


        # for each key I get the collection of summary nodes
        # I set up a filewriter for each summary node
        self.summary_nodes = {sk: tf.get_collection(sk) for sk in self.summary_keys}

        for sk in self.summary_keys:
            self.summary_writers[sk] = [tf.compat.v1.summary.FileWriter(self._tensorboard_dir+sn.name)
                                        for sn in self.summary_nodes[sk]]


    def create_hooks(self, config):

