fix: Add filterwarnings for jax.xla_computation DeprecationWarning (#2523)

matthewfeickert · web-flow · commit 7259d60ba6f4 · 2024-06-20T00:27:58.000-04:00
* Add an ignore to filterwarnings to avoid a DeprecationWarning on jax.xla_computation in jax v0.4.30+. > DeprecationWarning: jax.xla_computation is deprecated. > Please use the AOT APIs. - c.f. jax-ml/jax#21923
diff --git a/pyproject.toml b/pyproject.toml
@@ -227,6 +227,7 @@ filterwarnings = [
     "ignore:jsonschema.RefResolver is deprecated as of v4.18.0, in favor of the:DeprecationWarning",  # Issue #2139
     "ignore:Skipping device Apple Paravirtual device that does not support Metal 2.0:UserWarning",  # Can't fix given hardware/virtualized device
     'ignore:Type google._upb._message.[A-Z]+ uses PyType_Spec with a metaclass that has custom:DeprecationWarning',  # protobuf via tensorflow
+    "ignore:jax.xla_computation is deprecated. Please use the AOT APIs:DeprecationWarning",  # jax v0.4.30
 ]
 
 [tool.coverage.run]

Original file line number	Diff line number	Diff line change
`@@ -227,6 +227,7 @@ filterwarnings = [`
`227`	`227`	`"ignore:jsonschema.RefResolver is deprecated as of v4.18.0, in favor of the:DeprecationWarning", # Issue #2139`
`228`	`228`	`"ignore:Skipping device Apple Paravirtual device that does not support Metal 2.0:UserWarning", # Can't fix given hardware/virtualized device`
`229`	`229`	`'ignore:Type google._upb._message.[A-Z]+ uses PyType_Spec with a metaclass that has custom:DeprecationWarning', # protobuf via tensorflow`
	`230`	`+ "ignore:jax.xla_computation is deprecated. Please use the AOT APIs:DeprecationWarning", # jax v0.4.30`
`230`	`231`	`]`
`231`	`232`
`232`	`233`	`[tool.coverage.run]`