Decorator marking a test that requires JAX & Flax. These tests are skipped when one / both are not installed
(test_case)
| 468 | |
| 469 | |
| 470 | def require_flax(test_case): |
| 471 | """ |
| 472 | Decorator marking a test that requires JAX & Flax. These tests are skipped when one / both are not installed |
| 473 | """ |
| 474 | return unittest.skipUnless(is_flax_available(), "test requires JAX & Flax")(test_case) |
| 475 | |
| 476 | |
| 477 | def require_compel(test_case): |
nothing calls this directly
no test coverage detected
searching dependent graphs…