From 0e43709a3077fcc947b42ff02224a6ad8daf3403 Mon Sep 17 00:00:00 2001
From: Viraj Chaudhari <72896239+virajvchaudhari@users.noreply.github.com>
Date: Tue, 29 Aug 2023 14:43:50 -0700
Subject: [PATCH] infra: update braket dependencies

---
 base/jobs/docker/1.0/py3/Dockerfile.cpu        | 12 ++++++------
 pytorch/jobs/docker/2.0/py3/Dockerfile.gpu     | 12 ++++++------
 tensorflow/jobs/docker/2.12/py3/Dockerfile.gpu | 12 ++++++------
 3 files changed, 18 insertions(+), 18 deletions(-)

diff --git a/base/jobs/docker/1.0/py3/Dockerfile.cpu b/base/jobs/docker/1.0/py3/Dockerfile.cpu
index 14d80a9..7d54daf 100644
--- a/base/jobs/docker/1.0/py3/Dockerfile.cpu
+++ b/base/jobs/docker/1.0/py3/Dockerfile.cpu
@@ -107,10 +107,10 @@ RUN apt-get update && apt-get -y install cmake protobuf-compiler
 
 # Installing our custom python libraries
 RUN ${PIP} install --no-cache --upgrade \
-        amazon-braket-default-simulator==1.18.1 \
-        amazon-braket-schemas==1.18.0 \
-        amazon-braket-pennylane-plugin==1.18.0 \
-        amazon-braket-sdk==1.50.0 \
+        amazon-braket-default-simulator==1.20.0 \
+        amazon-braket-schemas==1.19.1 \
+        amazon-braket-pennylane-plugin==1.20.1 \
+        amazon-braket-sdk==1.54.2 \
         awscli==1.29.4 \
         botocore==1.31.4 \
         boto3==1.28.3 \
@@ -118,8 +118,8 @@ RUN ${PIP} install --no-cache --upgrade \
         matplotlib==3.6.3 \
         numpy==1.23.5 \
         pandas==2.0.0 \
-        pennylane==0.30.0 \
-        PennyLane-Lightning==0.30.0 \
+        pennylane==0.31.0 \
+        PennyLane-Lightning==0.31.0 \
         scikit-learn==1.2.2 \
         requests==2.26.0 \
         scipy==1.9.3
diff --git a/pytorch/jobs/docker/2.0/py3/Dockerfile.gpu b/pytorch/jobs/docker/2.0/py3/Dockerfile.gpu
index a3eea18..3edc04d 100644
--- a/pytorch/jobs/docker/2.0/py3/Dockerfile.gpu
+++ b/pytorch/jobs/docker/2.0/py3/Dockerfile.gpu
@@ -73,10 +73,10 @@ RUN apt-get update && apt-get -y install cmake protobuf-compiler
 
 # Installing our custom python libraries
 RUN ${PIP} install --no-cache --upgrade \
-        amazon-braket-default-simulator==1.18.1 \
-        amazon-braket-schemas==1.18.0 \
-        amazon-braket-pennylane-plugin==1.18.0 \
-        amazon-braket-sdk==1.50.0 \
+        amazon-braket-default-simulator==1.20.0 \
+        amazon-braket-schemas==1.19.1 \
+        amazon-braket-pennylane-plugin==1.20.1 \
+        amazon-braket-sdk==1.54.2 \
         awscli==1.29.4 \
         botocore==1.31.4 \
         boto3==1.28.3 \
@@ -91,8 +91,8 @@ RUN ${PIP} install --no-cache --upgrade \
         numpy==1.23.5 \
         openfermion==1.5.1 \
         pandas==2.0.0 \
-        PennyLane==0.30.0 \
-        PennyLane-Lightning[gpu]==0.30.0 \
+        PennyLane==0.31.0 \
+        PennyLane-Lightning[gpu]==0.31.0 \
         pydantic==1.10.6 \
         requests==2.26.0 \
         scikit-learn==1.2.2 \
diff --git a/tensorflow/jobs/docker/2.12/py3/Dockerfile.gpu b/tensorflow/jobs/docker/2.12/py3/Dockerfile.gpu
index 9022e09..89a1eee 100644
--- a/tensorflow/jobs/docker/2.12/py3/Dockerfile.gpu
+++ b/tensorflow/jobs/docker/2.12/py3/Dockerfile.gpu
@@ -15,10 +15,10 @@ RUN apt-get update \
 
 # Installing our custom python libraries
 RUN ${PIP} install --no-cache --upgrade \
-        amazon-braket-default-simulator==1.18.1 \
-        amazon-braket-schemas==1.18.0 \
-        amazon-braket-pennylane-plugin==1.18.0 \
-        amazon-braket-sdk==1.50.0 \
+        amazon-braket-default-simulator==1.20.0 \
+        amazon-braket-schemas==1.19.1 \
+        amazon-braket-pennylane-plugin==1.20.1 \
+        amazon-braket-sdk==1.54.2 \
         awscli==1.29.4 \
         botocore==1.31.4 \
         boto3==1.28.3 \
@@ -32,8 +32,8 @@ RUN ${PIP} install --no-cache --upgrade \
         numpy==1.23.5 \
         openfermion==1.5.1 \
         pandas==2.0.0 \
-        PennyLane==0.30.0 \
-        PennyLane-Lightning==0.30.0 \
+        PennyLane==0.31.0 \
+        PennyLane-Lightning==0.31.0 \
         pydantic==1.10.6 \
         requests==2.26.0 \
         scikit-learn==1.2.2 \