From d9fb969489965f69722ab20328310f485d8e7814 Mon Sep 17 00:00:00 2001
From: Rebecca Chen <rechen@google.com>
Date: Mon, 22 Mar 2021 17:52:39 +0000
Subject: [PATCH] Internal change

PiperOrigin-RevId: 364357405
---
 glassy_dynamics/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/glassy_dynamics/train.py b/glassy_dynamics/train.py
index 590777a..190e617 100644
--- a/glassy_dynamics/train.py
+++ b/glassy_dynamics/train.py
@@ -180,7 +180,7 @@ def train_model(train_file_pattern: Text,
                 mlp_kwargs: Optional[Dict[Text, Any]] = None,
                 edge_threshold: float = 2.0,
                 measurement_store_interval: int = 1000,
-                checkpoint_path: Optional[Text] = None) -> float:
+                checkpoint_path: Optional[Text] = None) -> float:  # pytype: disable=annotation-type-mismatch
   """Trains GraphModel using tensorflow.
 
   Args: