Updated the version to v0.0.8

PiperOrigin-RevId: 726291815

Author: lukebaumann

pathwaysutils/__init__.py

@@ -28,7 +28,7 @@
 
 # A new PyPI release will be pushed every time `__version__` is increased.
 # When changing this, also update the CHANGELOG.md.
-__version__ = "v0.0.7"
+__version__ = "v0.0.8"
 
 
 #  This is a brittle implementation since the platforms value is not necessarily

pathwaysutils/test/persistence_test.py

@@ -14,7 +14,11 @@ class PersistenceTest(absltest.TestCase):
   name = "name"
   dtype = np.dtype(np.int32)
   shape = [8, 4]
-  timeout = datetime.timedelta(seconds=30)
+  timeout = datetime.timedelta(seconds=3)
+
+  def setUp(self):
+    jax.config.update("jax_platforms", "cpu")
+    super().setUp()
 
   def test_get_read_request(self):
     devices = jax.devices()

pathwaysutils/test/plugin_executable_test.py

@@ -13,6 +13,10 @@
 
 class PluginExecutableTest(absltest.TestCase):
 
+  def setUp(self):
+    jax.config.update("jax_platforms", "cpu")
+    super().setUp()
+
   def test_bad_json_program(self):
     with self.assertRaisesRegex(XlaRuntimeError, "INVALID_ARGUMENT"):
       PluginExecutable('{"printTextRequest":{"badParamName":"foo"}}')