From e69ebb8ef0c536734c40f936818f6b85fa18bcfd Mon Sep 17 00:00:00 2001 From: pseudo-rnd-thoughts Date: Mon, 12 Aug 2024 18:20:07 +0100 Subject: [PATCH] run isort --- envpool/python/xla_template.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/envpool/python/xla_template.py b/envpool/python/xla_template.py index ae934a12..6aad7e48 100644 --- a/envpool/python/xla_template.py +++ b/envpool/python/xla_template.py @@ -18,12 +18,11 @@ from typing import Any, Callable, List, Tuple, Union import numpy as np -from jax import core, dtypes +from jax import core, dtypes, interpreters from jax import numpy as jnp from jax.core import ShapedArray from jax.interpreters import xla from jax.lib import xla_client -from jax import interpreters def _shape_with_layout(