Remove now redundant mask type casting. Minor formatting fixes.

PiperOrigin-RevId: 602449563
google · Feb 1, 2024 · 06c8ad8 · 06c8ad8
1 parent f96f176
commit 06c8ad8
Show file tree

Hide file tree

Showing 2 changed files with 127 additions and 90 deletions.
diff --git a/neural_tangents/__init__.py b/neural_tangents/__init__.py
@@ -16,7 +16,7 @@
 """Public Neural Tangents modules and functions."""
 
 
-__version__ = '0.6.5'
+__version__ = '0.6.6'
 
 from . import experimental
 from . import predict