Fixed missing imports

STCYC · Feb 19, 2018 · a932300 · a932300
1 parent 39f4ea9
commit a932300
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/tensorflow/contrib/timeseries/python/timeseries/BUILD b/tensorflow/contrib/timeseries/python/timeseries/BUILD
@@ -140,6 +140,7 @@ py_library(
         "//tensorflow/python:framework_ops",
         "//tensorflow/python:math_ops",
         "//tensorflow/python:state_ops",
+        "//tensorflow/python:summary",
         "//tensorflow/python:util",
         "//tensorflow/python:variable_scope",
         "//tensorflow/python/estimator:estimator_py",

diff --git a/tensorflow/contrib/timeseries/python/timeseries/head.py b/tensorflow/contrib/timeseries/python/timeseries/head.py
@@ -36,6 +36,7 @@
 from tensorflow.python.ops import state_ops
 from tensorflow.python.ops import variable_scope
 from tensorflow.python.util import nest
+from tensorflow.python.summary import summary
 
 
 def time_series_regression_head(model,
@@ -84,7 +85,7 @@ def create_loss(self, features, mode, logits=None, labels=None):
       model_outputs = self.state_manager.define_loss(
           self.model, features, mode)
       summary.scalar(
-        head_lib._summary_key(self._name, metric_keys.LOSS),
+        head_lib._summary_key(self._name, metric_keys.MetricKeys.LOSS),
         model_outputs.loss)
     return model_outputs