Change NINF

PiperOrigin-RevId: 555650113 Change-Id: I36cf70cbfd560edd6173ad07d0688e55a9ad3539
google-deepmind · Aug 10, 2023 · d880f68 · d880f68
1 parent 655e123
commit d880f68
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pgmax/utils/__init__.py b/pgmax/utils/__init__.py
@@ -34,7 +34,7 @@
 # What value to use as a "base" value for, e.g., log potentials of specific
 # configs.  If NEG_INF is too close to MSG_NEG_INF, it'll blow up as unlikely
 # configs become extremely likely.
-NEG_INF = jnp.NINF
+NEG_INF = -jnp.inf
 
 
 def cached_property(func: Callable[..., Any]) -> property: