Test sym vis (#118)

Author: makslevental

.github/workflows/test.yml

@@ -96,6 +96,7 @@ jobs:
         shell: bash
         run: |
           export PIP_FIND_LINKS=https://makslevental.github.io/wheels
+          pip install .[test,mlir] -v
           HOST_MLIR_PYTHON_PACKAGE_PREFIX=jaxlib.mlir pip install .[test,jax] -v
 
       - name: Test

tests/test_other_hosts.py

@@ -5,6 +5,8 @@
 
 @pytest.mark.skipif(jax_not_installed(), reason="jax not installed")
 def test_jax_trampolines_smoke():
+    from mlir import ir
+    from jaxlib.mlir import ir
     # noinspection PyUnresolvedReferences
     from jaxlib.mlir.extras import context
     # noinspection PyUnresolvedReferences