Cosmetic changes.

PiperOrigin-RevId: 432217865
Change-Id: I2c6aa125e691b603fdf2f097876049366bb1d9aa

Author: Johannes Ballé

models/bls2017.py

@@ -36,8 +36,8 @@
 from absl import app
 from absl.flags import argparse_flags
 import tensorflow as tf
-import tensorflow_datasets as tfds
 import tensorflow_compression as tfc
+import tensorflow_datasets as tfds
 
 
 def read_png(filename):

models/bmshj2018.py

@@ -34,8 +34,8 @@
 from absl import app
 from absl.flags import argparse_flags
 import tensorflow as tf
-import tensorflow_datasets as tfds
 import tensorflow_compression as tfc
+import tensorflow_datasets as tfds
 
 
 def read_png(filename):

models/hific/archs.py

@@ -24,7 +24,6 @@
 from compare_gan.architectures import arch_ops
 import numpy as np
 import tensorflow.compat.v1 as tf
-
 import tensorflow_compression as tfc
 
 from hific.helpers import ModelMode

models/ms2020.py

@@ -35,8 +35,8 @@
 from absl import app
 from absl.flags import argparse_flags
 import tensorflow as tf
-import tensorflow_datasets as tfds
 import tensorflow_compression as tfc
+import tensorflow_datasets as tfds
 
 
 def read_png(filename):

models/toy_sources/ntc.py

@@ -3,8 +3,8 @@
 import matplotlib.pyplot as plt
 import numpy as np
 import tensorflow as tf
-import tensorflow_probability as tfp
 import tensorflow_compression as tfc
+import tensorflow_probability as tfp
 from toy_sources import compression_model
 
 tfpd = tfp.distributions

models/toy_sources/toy_sources.ipynb

@@ -58,8 +58,8 @@
         "from absl import logging\n",
         "import numpy as np\n",
         "import tensorflow as tf\n",
-        "import tensorflow_probability as tfp\n",
         "import tensorflow_compression as tfc\n",
+        "import tensorflow_probability as tfp\n",
         "\n",
         "tfm = tf.math\n",
         "tfkl = tf.keras.layers\n",