Skip to content

Commit

Permalink
fix: Add filterwarnings for jax.xla_computation DeprecationWarning (#…
Browse files Browse the repository at this point in the history
…2523)

* 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
  • Loading branch information
matthewfeickert authored Jun 20, 2024
1 parent 2e4b930 commit 7259d60
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -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]
Expand Down

0 comments on commit 7259d60

Please sign in to comment.