add missing numpy import · feibyte/docker-python@5dc100d · GitHub

feibyte / docker-python Public

forked from Kaggle/docker-python

• Notifications You must be signed in to change notification settings
• Fork 0
• Star 0

• Code
• Pull requests
• Actions
• Projects
• Security and quality
• Insights

Additional navigation options

Commit 5dc100d

Browse filesBrowse the repository at this point in the historyBrowse files

rosbo

committed

add missing numpy import

1 parent 393a824 commit 5dc100d
Copy full SHA for 5dc100d

1 file changed

+1-1Lines changed: 1 addition & 1 deletion

File tree

Expand file treeCollapse file tree

Open diff view settings

Filter options

• tests
  • test_jax.py

Expand file treeCollapse file tree

Open diff view settings

Collapse file

‎tests/test_jax.py‎

Copy file name to clipboardExpand all lines: tests/test_jax.py

+1-1Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
@@ -7,14 +7,14 @@
7	7
8	8	class TestJAX(unittest.TestCase):
9	9	def tanh(self, x):
	10	+ import jax.numpy as np
10	11	y = np.exp(-2.0 * x)
11	12	return (1.0 - y) / (1.0 + y)
12	13
13	14	@gpu_test
14	15	def test_JAX(self):
15	16	# importing inside the gpu-only test because these packages can't be
16	17	# imported on the CPU image since they are not present there.
17		- import jax.numpy as np
18	18	from jax import grad, jit
19	19
20	20	grad_tanh = grad(self.tanh)

0 commit comments

Comments

0 (0)