From f4d240c03607659e1d9006f8723e4862b819f412 Mon Sep 17 00:00:00 2001 From: Jake VanderPlas Date: Fri, 25 Mar 2022 15:02:45 -0700 Subject: [PATCH] Remove lax_numpy from jax.numpy namespace This is a private module that was inadvertently exported in the past. --- CHANGELOG.md | 1 + jax/numpy/__init__.py | 3 --- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 05df3bbe9..00dc3f946 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -14,6 +14,7 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK. * Changes: * added {func}`jax.random.loggamma` & improved behavior of {func}`jax.random.beta` and {func}`jax.random.dirichlet` for small parameter values ({jax-issue}`#9906`). + * the private `lax_numpy` submodule is no longer exposed in the `jax.numpy` namespace ({jax-issue}`#10029`). * Deprecations: * {func}`jax.nn.normalize` is being deprecated. Use {func}`jax.nn.standardize` instead ({jax-issue}`#9899`). diff --git a/jax/numpy/__init__.py b/jax/numpy/__init__.py index 82641a8db..e460e39cb 100644 --- a/jax/numpy/__init__.py +++ b/jax/numpy/__init__.py @@ -436,9 +436,6 @@ from jax._src.numpy.ufuncs import ( from jax._src.numpy.vectorize import vectorize as vectorize -# TODO(phawkins): remove this import after fixing users. -from jax._src.numpy import lax_numpy - # Module initialization is encapsulated in a function to avoid accidental # namespace pollution. def _init():