Minor cleaning.

jax-md · Dec 7, 2019 · 92b02e7 · 92b02e7
1 parent 65d0ecf
commit 92b02e7
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 8 deletions.
diff --git a/examples/minimize.py b/examples/minimize.py
@@ -26,7 +26,7 @@
 
 import jax.numpy as np
 from jax import jit
-from jax_md import space, energy, minimize, quantity, smap, partition
+from jax_md import space, energy, minimize, quantity, smap
 from jax_md.util import f32, i32
 
 def main(unused_argv):
@@ -76,11 +76,5 @@ def main(unused_argv):
       print('{:.2f}\t{:.2f}\t{:.2f}'.format(
           step, energy_fn(R), np.max(force_fn(R))))
 
-  verlet_list_fn = partition.verlet_list(metric, box_size, 2., R)
-  idx = verlet_list_fn(R)
-  print(idx.shape)
-  print(idx)
-  print(np.mean(idx == idx.shape[0]))
-
 if __name__ == '__main__':
   app.run(main)
diff --git a/tests/.#space_test.py b/tests/.#space_test.py