1 files changed, 40 insertions, 38 deletions
diff --git a/tensorflow/tensorboard/components/tf_tensorboard/autoReloadBehavior.ts b/tensorflow/tensorboard/components/tf_tensorboard/autoReloadBehavior.ts
index 1f6b4cf641..54df16f5b5 100644
--- a/tensorflow/tensorboard/components/tf_tensorboard/autoReloadBehavior.ts
+++ b/tensorflow/tensorboard/components/tf_tensorboard/autoReloadBehavior.ts
@@ -12,49 +12,51 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License.
 ==============================================================================*/
-module TF.TensorBoard {
-  export var AUTORELOAD_LOCALSTORAGE_KEY = 'TF.TensorBoard.autoReloadEnabled';
 
-  var getAutoReloadFromLocalStorage: () => boolean = () => {
-    var val = window.localStorage.getItem(AUTORELOAD_LOCALSTORAGE_KEY);
-    return val === 'true' || val == null;  // defaults to true
-  };
+export var AUTORELOAD_LOCALSTORAGE_KEY = 'TF.TensorBoard.autoReloadEnabled';
 
-  export var AutoReloadBehavior = {
-    properties: {
-      autoReloadEnabled: {
-        type: Boolean,
-        observer: '_autoReloadObserver',
-        value: getAutoReloadFromLocalStorage,
-      },
-      _autoReloadId: {
-        type: Number,
-      },
-      autoReloadIntervalSecs: {
-        type: Number,
-        value: 30,
-      },
+var getAutoReloadFromLocalStorage: () => boolean = () => {
+  var val = window.localStorage.getItem(AUTORELOAD_LOCALSTORAGE_KEY);
+  return val === 'true' || val == null;  // defaults to true
+};
+
+/**
+ * @polymerBehavior
+ */
+export var AutoReloadBehavior = {
+  properties: {
+    autoReloadEnabled: {
+      type: Boolean,
+      observer: '_autoReloadObserver',
+      value: getAutoReloadFromLocalStorage,
     },
-    detached: function() {
-      window.clearTimeout(this._autoReloadId);
+    _autoReloadId: {
+      type: Number,
     },
-    _autoReloadObserver: function(autoReload) {
-      window.localStorage.setItem(AUTORELOAD_LOCALSTORAGE_KEY, autoReload);
-      if (autoReload) {
-        var _this = this;
-        this._autoReloadId = window.setTimeout(
-            this._doAutoReload.bind(this), this.autoReloadIntervalSecs * 1000);
-      } else {
-        window.clearTimeout(this._autoReloadId);
-      }
+    autoReloadIntervalSecs: {
+      type: Number,
+      value: 30,
     },
-    _doAutoReload: function() {
-      if (this.reload == null) {
-        throw new Error('AutoReloadBehavior requires a reload method');
-      }
-      this.reload();
+  },
+  detached: function() {
+    window.clearTimeout(this._autoReloadId);
+  },
+  _autoReloadObserver: function(autoReload) {
+    window.localStorage.setItem(AUTORELOAD_LOCALSTORAGE_KEY, autoReload);
+    if (autoReload) {
+      var _this = this;
       this._autoReloadId = window.setTimeout(
           this._doAutoReload.bind(this), this.autoReloadIntervalSecs * 1000);
+    } else {
+      window.clearTimeout(this._autoReloadId);
+    }
+  },
+  _doAutoReload: function() {
+    if (this.reload == null) {
+      throw new Error('AutoReloadBehavior requires a reload method');
     }
-  };
-}
+    this.reload();
+    this._autoReloadId = window.setTimeout(
+        this._doAutoReload.bind(this), this.autoReloadIntervalSecs * 1000);
+  }
+};