Internal change

PiperOrigin-RevId: 364357405
liameihao · Mar 26, 2021 · d9fb969 · d9fb969
1 parent 53326fe
commit d9fb969
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: