Error with latest tensorflow

The logger has this error, probably due the fact that FileWriter has been deprecated. 
'tensorboard.summary._tf.summary' has no attribute 'FileWriter

Fixed by changing 'FileWriter' to'create_file_writer'

logger.py

@@ -6,9 +6,9 @@ class Logger(object):
 
     def __init__(self, log_dir):
         """Initialize summary writer."""
-        self.writer = tf.summary.FileWriter(log_dir)
+        self.writer = tf.summary.create_file_writer(log_dir)
 
     def scalar_summary(self, tag, value, step):
         """Add scalar summary."""
         summary = tf.Summary(value=[tf.Summary.Value(tag=tag, simple_value=value)])
-        self.writer.add_summary(summary, step)
+        self.writer.add_summary(summary, step)