From b040537942fff6c514c4c32ec817e59418010087 Mon Sep 17 00:00:00 2001 From: JerryChen97 Date: Fri, 17 Jul 2026 16:21:47 -0400 Subject: [PATCH 1/4] Clear the remaining 3.11 and probably fix braket as well --- .github/workflows/braket-latest-latest.yml | 4 ---- .github/workflows/braket-latest-rc.yml | 4 ---- .github/workflows/braket-latest-stable.yml | 6 +----- .github/workflows/braket-stable-latest.yml | 4 ---- .github/workflows/braket-stable-stable.yml | 6 +----- .github/workflows/ionq-latest-stable.yml | 2 +- .github/workflows/ionq-stable-stable.yml | 2 +- .github/workflows/qiskit-latest-stable.yml | 2 +- .github/workflows/qiskit-stable-stable.yml | 2 +- workflow-template-stable.yml | 2 +- 10 files changed, 7 insertions(+), 27 deletions(-) diff --git a/.github/workflows/braket-latest-latest.yml b/.github/workflows/braket-latest-latest.yml index 41860db..9da8963 100644 --- a/.github/workflows/braket-latest-latest.yml +++ b/.github/workflows/braket-latest-latest.yml @@ -38,10 +38,6 @@ jobs: uses: actions/setup-python@v2 with: python-version: "3.12" - - - name: Install TF - run: | - pip install tensorflow~=$TF_VERSION keras~=$TF_VERSION - name: Install JAX run: | diff --git a/.github/workflows/braket-latest-rc.yml b/.github/workflows/braket-latest-rc.yml index eaa76f5..98554d8 100644 --- a/.github/workflows/braket-latest-rc.yml +++ b/.github/workflows/braket-latest-rc.yml @@ -38,10 +38,6 @@ jobs: uses: actions/setup-python@v2 with: python-version: "3.12" - - - name: Install TF - run: | - pip install tensorflow~=$TF_VERSION keras~=$TF_VERSION - name: Install JAX run: | diff --git a/.github/workflows/braket-latest-stable.yml b/.github/workflows/braket-latest-stable.yml index d92a5df..3f0bc0f 100644 --- a/.github/workflows/braket-latest-stable.yml +++ b/.github/workflows/braket-latest-stable.yml @@ -37,11 +37,7 @@ jobs: - name: Set up Python uses: actions/setup-python@v2 with: - python-version: "3.11" - - - name: Install TF - run: | - pip install tensorflow~=$TF_VERSION keras~=$TF_VERSION + python-version: "3.12" - name: Install JAX run: | diff --git a/.github/workflows/braket-stable-latest.yml b/.github/workflows/braket-stable-latest.yml index d456a6c..27e967e 100644 --- a/.github/workflows/braket-stable-latest.yml +++ b/.github/workflows/braket-stable-latest.yml @@ -38,10 +38,6 @@ jobs: uses: actions/setup-python@v2 with: python-version: "3.12" - - - name: Install TF - run: | - pip install tensorflow~=$TF_VERSION keras~=$TF_VERSION - name: Install JAX run: | diff --git a/.github/workflows/braket-stable-stable.yml b/.github/workflows/braket-stable-stable.yml index 5d3f910..62783ce 100644 --- a/.github/workflows/braket-stable-stable.yml +++ b/.github/workflows/braket-stable-stable.yml @@ -37,11 +37,7 @@ jobs: - name: Set up Python uses: actions/setup-python@v2 with: - python-version: "3.11" - - - name: Install TF - run: | - pip install tensorflow~=$TF_VERSION keras~=$TF_VERSION + python-version: "3.12" - name: Install JAX run: | diff --git a/.github/workflows/ionq-latest-stable.yml b/.github/workflows/ionq-latest-stable.yml index 740fa24..fc27d0f 100644 --- a/.github/workflows/ionq-latest-stable.yml +++ b/.github/workflows/ionq-latest-stable.yml @@ -35,7 +35,7 @@ jobs: - name: Set up Python uses: actions/setup-python@v2 with: - python-version: "3.11" + python-version: "3.12" - name: Install requirements run: | diff --git a/.github/workflows/ionq-stable-stable.yml b/.github/workflows/ionq-stable-stable.yml index 8ec2bc3..73ccaa7 100644 --- a/.github/workflows/ionq-stable-stable.yml +++ b/.github/workflows/ionq-stable-stable.yml @@ -35,7 +35,7 @@ jobs: - name: Set up Python uses: actions/setup-python@v2 with: - python-version: "3.11" + python-version: "3.12" - name: Install requirements run: | diff --git a/.github/workflows/qiskit-latest-stable.yml b/.github/workflows/qiskit-latest-stable.yml index 9e6cf84..5bb2829 100644 --- a/.github/workflows/qiskit-latest-stable.yml +++ b/.github/workflows/qiskit-latest-stable.yml @@ -35,7 +35,7 @@ jobs: - name: Set up Python uses: actions/setup-python@v2 with: - python-version: "3.11" + python-version: "3.12" - name: Install requirements run: | diff --git a/.github/workflows/qiskit-stable-stable.yml b/.github/workflows/qiskit-stable-stable.yml index a01a8e3..5ca4a51 100644 --- a/.github/workflows/qiskit-stable-stable.yml +++ b/.github/workflows/qiskit-stable-stable.yml @@ -35,7 +35,7 @@ jobs: - name: Set up Python uses: actions/setup-python@v2 with: - python-version: "3.11" + python-version: "3.12" - name: Install requirements run: | diff --git a/workflow-template-stable.yml b/workflow-template-stable.yml index 37b8f36..8f752f7 100644 --- a/workflow-template-stable.yml +++ b/workflow-template-stable.yml @@ -40,7 +40,7 @@ jobs: - name: Set up Python uses: actions/setup-python@v2 with: - python-version: "3.11" + python-version: "3.12" {%- if additional_setup is defined %} {{ additional_setup | indent(6, True) }} From e32325f7e7c136f2f81f03687cc59284bcd563f5 Mon Sep 17 00:00:00 2001 From: JerryChen97 Date: Fri, 17 Jul 2026 16:30:58 -0400 Subject: [PATCH 2/4] update torch and jax --- compile.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/compile.py b/compile.py index 89af071..cb3827a 100644 --- a/compile.py +++ b/compile.py @@ -49,7 +49,7 @@ pip install jax==$JAX_VERSION jaxlib==$JAX_VERSION""" ), - "additional_env_vars": "TF_VERSION: 2.12.0\n TORCH_VERSION: 2.0.0+cpu\n JAX_VERSION: 0.4.28", + "additional_env_vars": "TF_VERSION: 2.12.0\n TORCH_VERSION: 2.9.1+cpu\n JAX_VERSION: 0.7.1", "no_deprecation_error": True, }, ] From a7ad8f9726ec35af0b100dec4a413fbf44192cc4 Mon Sep 17 00:00:00 2001 From: JerryChen97 Date: Fri, 17 Jul 2026 16:36:29 -0400 Subject: [PATCH 3/4] re-compile --- .github/workflows/braket-latest-latest.yml | 7 +++---- .github/workflows/braket-latest-rc.yml | 7 +++---- .github/workflows/braket-latest-stable.yml | 7 +++---- .github/workflows/braket-stable-latest.yml | 7 +++---- .github/workflows/braket-stable-stable.yml | 7 +++---- compile.py | 8 ++------ 6 files changed, 17 insertions(+), 26 deletions(-) diff --git a/.github/workflows/braket-latest-latest.yml b/.github/workflows/braket-latest-latest.yml index 9da8963..d115915 100644 --- a/.github/workflows/braket-latest-latest.yml +++ b/.github/workflows/braket-latest-latest.yml @@ -16,9 +16,8 @@ env: PLUGIN_BRANCH: main PLUGIN_PACKAGE: amazon-braket-pennylane-plugin PENNYLANE_BRANCH: main - TF_VERSION: 2.12.0 - TORCH_VERSION: 2.0.0+cpu - JAX_VERSION: 0.4.28 + TORCH_VERSION: 2.9.1+cpu + JAX_VERSION: 0.7.1 jobs: @@ -38,7 +37,7 @@ jobs: uses: actions/setup-python@v2 with: python-version: "3.12" - + - name: Install JAX run: | pip install jax==$JAX_VERSION jaxlib==$JAX_VERSION diff --git a/.github/workflows/braket-latest-rc.yml b/.github/workflows/braket-latest-rc.yml index 98554d8..d38fcd6 100644 --- a/.github/workflows/braket-latest-rc.yml +++ b/.github/workflows/braket-latest-rc.yml @@ -16,9 +16,8 @@ env: PLUGIN_BRANCH: main PLUGIN_PACKAGE: amazon-braket-pennylane-plugin PENNYLANE_BRANCH: main - TF_VERSION: 2.12.0 - TORCH_VERSION: 2.0.0+cpu - JAX_VERSION: 0.4.28 + TORCH_VERSION: 2.9.1+cpu + JAX_VERSION: 0.7.1 jobs: @@ -38,7 +37,7 @@ jobs: uses: actions/setup-python@v2 with: python-version: "3.12" - + - name: Install JAX run: | pip install jax==$JAX_VERSION jaxlib==$JAX_VERSION diff --git a/.github/workflows/braket-latest-stable.yml b/.github/workflows/braket-latest-stable.yml index 3f0bc0f..7d84367 100644 --- a/.github/workflows/braket-latest-stable.yml +++ b/.github/workflows/braket-latest-stable.yml @@ -16,9 +16,8 @@ env: PLUGIN_BRANCH: main PLUGIN_PACKAGE: amazon-braket-pennylane-plugin PENNYLANE_BRANCH: main - TF_VERSION: 2.12.0 - TORCH_VERSION: 2.0.0+cpu - JAX_VERSION: 0.4.28 + TORCH_VERSION: 2.9.1+cpu + JAX_VERSION: 0.7.1 jobs: @@ -38,7 +37,7 @@ jobs: uses: actions/setup-python@v2 with: python-version: "3.12" - + - name: Install JAX run: | pip install jax==$JAX_VERSION jaxlib==$JAX_VERSION diff --git a/.github/workflows/braket-stable-latest.yml b/.github/workflows/braket-stable-latest.yml index 27e967e..3d1ff17 100644 --- a/.github/workflows/braket-stable-latest.yml +++ b/.github/workflows/braket-stable-latest.yml @@ -16,9 +16,8 @@ env: PLUGIN_BRANCH: main PLUGIN_PACKAGE: amazon-braket-pennylane-plugin PENNYLANE_BRANCH: main - TF_VERSION: 2.12.0 - TORCH_VERSION: 2.0.0+cpu - JAX_VERSION: 0.4.28 + TORCH_VERSION: 2.9.1+cpu + JAX_VERSION: 0.7.1 jobs: @@ -38,7 +37,7 @@ jobs: uses: actions/setup-python@v2 with: python-version: "3.12" - + - name: Install JAX run: | pip install jax==$JAX_VERSION jaxlib==$JAX_VERSION diff --git a/.github/workflows/braket-stable-stable.yml b/.github/workflows/braket-stable-stable.yml index 62783ce..f2487df 100644 --- a/.github/workflows/braket-stable-stable.yml +++ b/.github/workflows/braket-stable-stable.yml @@ -16,9 +16,8 @@ env: PLUGIN_BRANCH: main PLUGIN_PACKAGE: amazon-braket-pennylane-plugin PENNYLANE_BRANCH: main - TF_VERSION: 2.12.0 - TORCH_VERSION: 2.0.0+cpu - JAX_VERSION: 0.4.28 + TORCH_VERSION: 2.9.1+cpu + JAX_VERSION: 0.7.1 jobs: @@ -38,7 +37,7 @@ jobs: uses: actions/setup-python@v2 with: python-version: "3.12" - + - name: Install JAX run: | pip install jax==$JAX_VERSION jaxlib==$JAX_VERSION diff --git a/compile.py b/compile.py index cb3827a..7071a33 100644 --- a/compile.py +++ b/compile.py @@ -39,17 +39,13 @@ "--device=braket.local.qubit --tb=short --skip-ops -k 'not Sample and not no_0_shots'", ], "tests_loc": "test/unit_tests", - "additional_setup": dedent(""" - - name: Install TF - run: | - pip install tensorflow~=$TF_VERSION keras~=$TF_VERSION - + "additional_setup": dedent(""" - name: Install JAX run: | pip install jax==$JAX_VERSION jaxlib==$JAX_VERSION""" ), - "additional_env_vars": "TF_VERSION: 2.12.0\n TORCH_VERSION: 2.9.1+cpu\n JAX_VERSION: 0.7.1", + "additional_env_vars": "TORCH_VERSION: 2.9.1+cpu\n JAX_VERSION: 0.7.1", "no_deprecation_error": True, }, ] From 4004af48b0a5ab5f5ca0a3e1c0950c454bdfd219 Mon Sep 17 00:00:00 2001 From: JerryChen97 Date: Fri, 17 Jul 2026 16:47:43 -0400 Subject: [PATCH 4/4] trigger