Added type annotations throughout.

l1zp · Oct 12, 2020 · 87528f7 · 87528f7
1 parent c4087ad
commit 87528f7
Show file tree

Hide file tree

Showing 17 changed files with 980 additions and 640 deletions.
diff --git a/.gitignore b/.gitignore
@@ -5,6 +5,7 @@
 /bazel-*
 .bazelrc
 .pytype/*
+**/.pytype/*
 /tensorflow
 .DS_Store
 build/

diff --git a/jax_md/__init__.py b/jax_md/__init__.py
@@ -20,3 +20,5 @@
 from jax_md import partition
 from jax_md import dataclasses
 from jax_md import nn
+from jax_md import interpolate
+from jax_md import util
diff --git a/jax_md/dataclasses.py b/jax_md/dataclasses.py
@@ -70,3 +70,4 @@ def static_field():
 
 replace = dataclasses.replace
 asdict = dataclasses.asdict
+astuple = dataclasses.astuple
Original file line number	Diff line number	Diff line change
Expand Up		@@ -70,3 +70,4 @@ def static_field():

		replace = dataclasses.replace
		asdict = dataclasses.asdict
		astuple = dataclasses.astuple