[envpool] Fix xla template import order

Author: Trinkle23897

envpool/python/xla_template.py

@@ -20,9 +20,9 @@
 import numpy as np
 from jax import core, dtypes
 from jax import numpy as jnp
+from jax._src.lib import xla_client
 from jax.core import ShapedArray
 from jax.interpreters import xla
-from jax._src.lib import xla_client
 
 _core = cast(Any, core)
 _xla = cast(Any, xla)