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
+ 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."""