Skip to content

Commit

Permalink
fix: revert integ test changes (#722)
Browse files Browse the repository at this point in the history
  • Loading branch information
ajberdy authored Oct 4, 2023
1 parent 5c1e4e7 commit 3d62bec
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions test/integ_tests/job_test_script.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,19 +11,19 @@
# ANY KIND, either express or implied. See the License for the specific
# language governing permissions and limitations under the License.

import json
import os

from braket.aws import AwsDevice
from braket.circuits import Circuit
from braket.jobs import (
get_hyperparameters,
get_job_device_arn,
save_job_checkpoint,
save_job_result,
)
from braket.jobs import save_job_checkpoint, save_job_result
from braket.jobs_data import PersistedJobDataFormat


def start_here():
hyperparameters = get_hyperparameters()
hp_file = os.environ["AMZN_BRAKET_HP_FILE"]
with open(hp_file, "r") as f:
hyperparameters = json.load(f)

if hyperparameters["test_case"] == "completed":
completed_job_script()
Expand All @@ -40,7 +40,7 @@ def completed_job_script():
print("Test job started!!!!!")

# Use the device declared in the Orchestration Script
device = AwsDevice(get_job_device_arn())
device = AwsDevice(os.environ["AMZN_BRAKET_DEVICE_ARN"])

bell = Circuit().h(0).cnot(0, 1)
for count in range(5):
Expand Down

0 comments on commit 3d62bec

Please sign in to comment.