From ce7347a52bb752c362e94214e5a8eb95cde7254a Mon Sep 17 00:00:00 2001 From: Ruturaj4 Date: Wed, 9 Apr 2025 19:36:15 -0500 Subject: [PATCH] Fix circular import in pallas core file --- jax/_src/pallas/mosaic/core.py | 1 - 1 file changed, 1 deletion(-) diff --git a/jax/_src/pallas/mosaic/core.py b/jax/_src/pallas/mosaic/core.py index 82fe9c2ba..bf6899fbe 100644 --- a/jax/_src/pallas/mosaic/core.py +++ b/jax/_src/pallas/mosaic/core.py @@ -28,7 +28,6 @@ from jax._src import core as jax_core from jax._src import dtypes from jax._src import util from jax._src.pallas import core as pallas_core -from jax._src.pallas import pallas_call import jax.numpy as jnp import numpy as np