Test pypi catalyst

.github/workflows/install_deps/action.yml

@@ -36,6 +36,11 @@ inputs:
     description: Indicate if PennyLane-Lightning should be installed from the master branch
     required: false
     default: 'true'
+  install_catalyst_test_pypi:
+    description: Indicate if catalyst should be installed from the test pypi location
+    required: false
+    type: boolean
+    default: 'false'
   additional_pip_packages:
     description: Additional packages to install. Values will be passed to pip install {value}
     required: false
@@ -100,6 +105,11 @@ runs:
       if: inputs.install_pennylane_lightning_master == 'true'
       run: pip install -i https://test.pypi.org/simple/ PennyLane-Lightning --pre --upgrade
 
+    - name: Install Catalyst from Test Pypi
+      shell: bash
+      if: inputs.install_catalyst_test_pypi == 'true'
+      run: pip install -i https://test.pypi.org/simple/ PennyLane-Catalyst==0.7.0.dev0 --upgrade;
+
     - name: Freeze dependencies
       shell: bash
       if: inputs.requirements_file != ''

.github/workflows/interface-unit-tests.yml

@@ -335,6 +335,39 @@ jobs:
       requirements_file: ${{ strategy.job-index == 0 && 'all_interfaces.txt' || '' }}
       disable_new_opmath: ${{ inputs.disable_new_opmath }}
 
+  catalyst-tests:
+    needs:
+      - setup-ci-load
+    strategy:
+      max-parallel: >-
+        ${{
+           fromJSON(needs.setup-ci-load.outputs.matrix-max-parallel).catalyst-tests
+           || fromJSON(needs.setup-ci-load.outputs.matrix-max-parallel).default
+         }}
+      matrix:
+        python-version: >-
+          ${{
+            fromJSON(needs.setup-ci-load.outputs.python-version).catalyst-tests
+            || fromJSON(needs.setup-ci-load.outputs.python-version).default
+           }}
+    if: ${{ !contains(fromJSON(needs.setup-ci-load.outputs.jobs-to-skip), 'catalyst-tests') }}
+    uses: ./.github/workflows/unit-test.yml
+    with:
+      job_name: catalyst-tests (${{ matrix.python-version }})
+      branch: ${{ inputs.branch }}
+      coverage_artifact_name: catalyst-tests-coverage
+      python_version: ${{ matrix.python-version }}
+      pipeline_mode: ${{ inputs.pipeline_mode }}
+      install_jax: true
+      install_tensorflow: false
+      install_pytorch: false
+      install_pennylane_lightning_master: true
+      install_catalyst_test_pypi: true
+      pytest_coverage_flags: ${{ inputs.pytest_coverage_flags }}
+      pytest_markers: catalyst
+      additional_pip_packages: matplotlib
+      requirements_file: ${{ strategy.job-index == 0 && 'catalyst.txt' || '' }}
+      disable_new_opmath: ${{ inputs.disable_new_opmath }}
 
   external-libraries-tests:
     needs:

.github/workflows/unit-test.yml

@@ -64,6 +64,11 @@ on:
         required: false
         type: boolean
         default: true
+      install_catalyst_test_pypi:
+        description: Indicate if catalyst should be installed from the test pypi location
+        required: false
+        type: boolean
+        default: false
       pytest_test_directory:
         description: The directory where the PennyLane tests are that should be run by PyTest
         required: false
@@ -148,6 +153,7 @@ jobs:
           install_jax: ${{ inputs.install_jax }}
           additional_pip_packages: ${{ inputs.additional_pip_packages }}
           install_pennylane_lightning_master: ${{ inputs.install_pennylane_lightning_master }}
+          install_catalyst_test_pypi: ${{ inputs.install_catalyst_test_pypi }}
           requirements_file: ${{ inputs.requirements_file }}
 
       - name: Set PyTest Args

tests/drawer/test_draw_catalyst.py

@@ -20,7 +20,7 @@
 catalyst = pytest.importorskip("catalyst")
 mpl = pytest.importorskip("matplotlib")
 
-pytestmark = pytest.mark.external
+pytestmark = pytest.mark.catalyst
 
 
 class TestCatalystDraw:

tests/ops/functions/test_matrix.py

@@ -660,7 +660,6 @@ def circuit(theta):
         assert np.allclose(matrix, expected_matrix)
 
     @pytest.mark.catalyst
-    @pytest.mark.external
     def test_catalyst(self):
         """Test with Catalyst interface"""
 

tests/test_compiler.py

@@ -29,7 +29,7 @@
 catalyst = pytest.importorskip("catalyst")
 jax = pytest.importorskip("jax")
 
-pytestmark = pytest.mark.external
+pytestmark = pytest.mark.catalyst
 
 from jax import numpy as jnp  # pylint:disable=wrong-import-order, wrong-import-position
 from jax.core import ShapedArray  # pylint:disable=wrong-import-order, wrong-import-position

tests/transforms/core/test_transform_dispatcher.py

@@ -224,7 +224,6 @@ class TestTransformDispatcher:  # pylint: disable=too-many-public-methods
     """Test the transform function (validate and dispatch)."""
 
     @pytest.mark.catalyst
-    @pytest.mark.external
     def test_error_on_qjit(self):
         """Test that an error is raised on when applying a transform to a qjit object."""