Suppress some pytype errors related to improved JAX types under pytype.

google / uncertainty-baselines

High-quality implementations of standard and SOTA methods on a variety of tasks.

Apache License 2.0

1.46k stars 203 forks source link

Closed copybara-service[bot] closed 1 year ago

copybara-service[bot] commented 1 year ago

Suppress some pytype errors related to improved JAX types under pytype.