MCPcopy Create free account
hub / github.com/Kaggle/docker-python / test_backend

Method test_backend

tests/test_jax.py:23–25  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

21 self.assertEqual(0.4199743, ag)
22
23 def test_backend(self):
24 expected_backend = 'cpu' if not isGPU() else 'gpu'
25 self.assertEqual(expected_backend, jax.default_backend())

Callers

nothing calls this directly

Calls 1

isGPUFunction · 0.90

Tested by

no test coverage detected