Update glass_dynamics to Python 3 imports.

PiperOrigin-RevId: 328363319

glassy_dynamics/apply_binary.py

@@ -14,11 +14,6 @@
 
 """Applies a graph-based network to predict particle mobilities in glasses."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import google_type_annotations
-from __future__ import print_function
-
 import os
 
 from absl import app

glassy_dynamics/graph_model.py

@@ -18,12 +18,8 @@ The architecture and performance of this model is described in our publication:
 "Unveiling the predictive power of static structure in glassy systems".
 """
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import google_type_annotations
-from __future__ import print_function
-
 import functools
+from typing import Any, Dict, Text, Tuple, Optional
 
 from graph_nets import graphs
 from graph_nets import modules as gn_modules
@@ -31,7 +27,6 @@ from graph_nets import utils_tf
 
 import sonnet as snt
 import tensorflow.compat.v1 as tf
-from typing import Any, Dict, Text, Tuple, Optional
 
 
 def make_graph_from_static_structure(

glassy_dynamics/graph_model_test.py

@@ -14,10 +14,6 @@
 
 """Tests for graph_model."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import itertools
 from absl.testing import parameterized
 

glassy_dynamics/train.py

@@ -14,22 +14,18 @@
 
 """Training pipeline for the prediction of particle mobilities in glasses."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import google_type_annotations
-from __future__ import print_function
 
 import collections
+import enum
 import pickle
+from typing import Any, Dict, List, Optional, Text, Tuple, Sequence
 
 from absl import logging
-import enum
 import numpy as np
 
 import tensorflow.compat.v1 as tf
 import tensorflow_probability as tfp
 
-from typing import Any, Dict, List, Optional, Text, Tuple, Sequence
 
 from glassy_dynamics import graph_model
 

glassy_dynamics/train_binary.py

@@ -14,11 +14,6 @@
 
 """Trains a graph-based network to predict particle mobilities in glasses."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import google_type_annotations
-from __future__ import print_function
-
 import os
 
 from absl import app

glassy_dynamics/train_test.py

@@ -14,10 +14,6 @@
 
 """Tests for train."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import os
 
 import numpy as np