from lapjax.experimental.jax2tf.tests import converters as converters
from lapjax.experimental.jax2tf.tests import call_tf_test as call_tf_test
from lapjax.experimental.jax2tf.tests import jax_primitives_coverage_test as jax_primitives_coverage_test
from lapjax.experimental.jax2tf.tests import jax2tf_test as jax2tf_test
from lapjax.experimental.jax2tf.tests import shape_poly_test as shape_poly_test
from lapjax.experimental.jax2tf.tests import control_flow_ops_test as control_flow_ops_test
from lapjax.experimental.jax2tf.tests import sharding_test as sharding_test
from lapjax.experimental.jax2tf.tests import models_test_main as models_test_main
from lapjax.experimental.jax2tf.tests import primitives_test as primitives_test
from lapjax.experimental.jax2tf.tests import model_harness as model_harness
from lapjax.experimental.jax2tf.tests import tf_test_util as tf_test_util
from lapjax.experimental.jax2tf.tests import jax2tf_limitations as jax2tf_limitations
from lapjax.experimental.jax2tf.tests import savedmodel_test as savedmodel_test
from lapjax.experimental.jax2tf.tests import primitive_harness as primitive_harness
import sys, importlib
from lapjax.lapsrc.wrapper import _wrap_module
_wrap_module(importlib.import_module(__name__.replace('lapjax', 'jax')), 
             sys.modules[__name__])
