Reorder imports

PiperOrigin-RevId: 852649279

Author: lukebaumann

pathwaysutils/test/debug/timing_test.py

@@ -17,10 +17,9 @@
 import time
 from unittest import mock
 
-from pathwaysutils.debug import timing
-
 from absl.testing import absltest
 from absl.testing import parameterized
+from pathwaysutils.debug import timing
 
 
 class TimingTest(parameterized.TestCase):

pathwaysutils/test/debug/watchdog_test.py

@@ -19,11 +19,9 @@
 import traceback
 from unittest import mock
 
-from pathwaysutils.debug import watchdog
-
 from absl.testing import absltest
 from absl.testing import parameterized
-
+from pathwaysutils.debug import watchdog
 
 class WatchdogTest(parameterized.TestCase):
   def test_watchdog_start_join(self):

pathwaysutils/test/initialize_test.py

@@ -14,11 +14,10 @@
 
 import os
 
-import jax
-from pathwaysutils import _initialize
-
 from absl.testing import absltest
 from absl.testing import parameterized
+import jax
+from pathwaysutils import _initialize
 
 
 class InitializeTest(parameterized.TestCase):

pathwaysutils/test/lru_cache_test.py

@@ -12,9 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from absl.testing import absltest
 import jax.extend
 from pathwaysutils import lru_cache
-from absl.testing import absltest
 
 
 class LruCacheTest(absltest.TestCase):

pathwaysutils/test/persistence_test.py

@@ -2,12 +2,11 @@
 
 import datetime
 
+from absl.testing import absltest
 import jax
 import numpy as np
 from pathwaysutils.persistence import helper
 
-from absl.testing import absltest
-
 
 class PersistenceTest(absltest.TestCase):
   location = "/path/to/location"

pathwaysutils/test/plugin_executable_test.py

@@ -3,9 +3,9 @@
 These should not exercise a specific feature that uses side channel, but rather
 the general logic of the class.
 """
+from absl.testing import absltest
 import jax
 from pathwaysutils import plugin_executable
-from absl.testing import absltest
 
 PluginExecutable = plugin_executable.PluginExecutable
 XlaRuntimeError = jax.errors.JaxRuntimeError

pathwaysutils/test/profiling_test.py

@@ -16,13 +16,12 @@
 import logging
 from unittest import mock
 
+from absl.testing import absltest
+from absl.testing import parameterized
 import jax
 from pathwaysutils import profiling
 import requests
 
-from absl.testing import absltest
-from absl.testing import parameterized
-
 
 class ProfilingTest(parameterized.TestCase):
   """Tests for Pathways on Cloud profiling."""

pathwaysutils/test/proxy_backend_test.py

@@ -15,13 +15,12 @@
 
 from unittest import mock
 
+from absl.testing import absltest
 import jax
 from jax.extend import backend
 from pathwaysutils import jax as pw_jax
 from pathwaysutils import proxy_backend
 
-from absl.testing import absltest
-
 
 class ProxyBackendTest(absltest.TestCase):