Mirrors_Framework/deepmind-research
1
0
Fork 0
mirror of https://github.com/google-deepmind/deepmind-research.git synced 2026-05-31 13:05:40 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update glass_dynamics to Python 3 imports.

Browse Source 
PiperOrigin-RevId: 328363319
This commit is contained in:
Florent Altché
2020-08-25 19:00:24 +01:00
committed by Diego de Las Casas
parent 8457046b2c
commit 486d8caf21
6 changed files with 3 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
glassy_dynamics/apply_binary.py
-5
View File
@@ -14,11 +14,6 @@
"""Applies a graph-based network to predict particle mobilities in glasses.""" """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 import os
from absl import app from absl import app
glassy_dynamics/graph_model.py
+1 -6
View File
@@ -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". "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 import functools
from typing import Any, Dict, Text, Tuple, Optional
from graph_nets import graphs from graph_nets import graphs
from graph_nets import modules as gn_modules from graph_nets import modules as gn_modules
@@ -31,7 +27,6 @@ from graph_nets import utils_tf
import sonnet as snt import sonnet as snt
import tensorflow.compat.v1 as tf import tensorflow.compat.v1 as tf
from typing import Any, Dict, Text, Tuple, Optional
def make_graph_from_static_structure( def make_graph_from_static_structure(
glassy_dynamics/graph_model_test.py
-4
View File
@@ -14,10 +14,6 @@
"""Tests for graph_model.""" """Tests for graph_model."""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import itertools import itertools
from absl.testing import parameterized from absl.testing import parameterized
glassy_dynamics/train.py
+2 -6
View File
@@ -14,22 +14,18 @@
"""Training pipeline for the prediction of particle mobilities in glasses.""" """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 collections
import enum
import pickle import pickle
from typing import Any, Dict, List, Optional, Text, Tuple, Sequence
from absl import logging from absl import logging
import enum
import numpy as np import numpy as np
import tensorflow.compat.v1 as tf import tensorflow.compat.v1 as tf
import tensorflow_probability as tfp import tensorflow_probability as tfp
from typing import Any, Dict, List, Optional, Text, Tuple, Sequence
from glassy_dynamics import graph_model from glassy_dynamics import graph_model
glassy_dynamics/train_binary.py
-5
View File
@@ -14,11 +14,6 @@
"""Trains a graph-based network to predict particle mobilities in glasses.""" """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 import os
from absl import app from absl import app
glassy_dynamics/train_test.py
-4
View File
@@ -14,10 +14,6 @@
"""Tests for train.""" """Tests for train."""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import os import os
import numpy as np import numpy as np