run isort

sail-sg · Aug 12, 2024 · e69ebb8 · e69ebb8
1 parent b7b8a49
commit e69ebb8
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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(