From 1119a77e1cfbbaa1fda7e7545c5a535b834c7c30 Mon Sep 17 00:00:00 2001 From: "changhao@mit.edu" Date: Thu, 13 Feb 2025 01:55:47 -0500 Subject: [PATCH] minor modifications on cli.py and run.py --- metriq_gym/cli.py | 2 -- metriq_gym/run.py | 4 ++-- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/metriq_gym/cli.py b/metriq_gym/cli.py index 372dc5c..50da96c 100644 --- a/metriq_gym/cli.py +++ b/metriq_gym/cli.py @@ -9,7 +9,6 @@ LIST_JOBS_HEADERS = ["ID", "Provider", "Device", "Type", "Dispatch time (UTC)"] - def list_jobs(jobs: list[MetriqGymJob], show_index: bool = True) -> None: """List jobs recorded in the job manager. @@ -30,7 +29,6 @@ def list_jobs(jobs: list[MetriqGymJob], show_index: bool = True) -> None: ) ) - def parse_arguments() -> argparse.Namespace: """ Parse command-line arguments for the quantum volume benchmark. diff --git a/metriq_gym/run.py b/metriq_gym/run.py index 4883ffe..525a1b3 100644 --- a/metriq_gym/run.py +++ b/metriq_gym/run.py @@ -58,7 +58,6 @@ def dispatch_job(args: argparse.Namespace, job_manager: JobManager) -> None: def poll_job(args: argparse.Namespace, job_manager: JobManager) -> None: - logger.info("Polling job...") if not args.job_id: jobs = job_manager.get_jobs() if not jobs: @@ -72,11 +71,12 @@ def poll_job(args: argparse.Namespace, job_manager: JobManager) -> None: if 0 <= selected_index < len(jobs): break else: - print("Invalid index. Please enter a number between 0 and", len(jobs) - 1) + print(f"Invalid index. Please enter a number between 0 and {len(jobs) - 1}") except ValueError: print("Invalid input. Please enter a valid number.") args.job_id = jobs[selected_index].id + logger.info("Polling job...") metriq_job: MetriqGymJob = job_manager.get_job(args.job_id) job_type: JobType = JobType(metriq_job.job_type) job_data: BenchmarkData = setup_job_data_class(job_type)(**metriq_job.data)