From ab5c1473f1bae21543f525aba15c5e182b9fe7b2 Mon Sep 17 00:00:00 2001 From: WrathfulSpatula Date: Fri, 7 Mar 2025 13:27:23 -0500 Subject: [PATCH] Fix linting --- metriq_gym/run.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/metriq_gym/run.py b/metriq_gym/run.py index 3e91aec..5e51d00 100644 --- a/metriq_gym/run.py +++ b/metriq_gym/run.py @@ -13,6 +13,8 @@ from metriq_gym.benchmarks import BENCHMARK_DATA_CLASSES, BENCHMARK_HANDLERS from metriq_gym.benchmarks.benchmark import Benchmark, BenchmarkData, BenchmarkResult from metriq_gym.cli import list_jobs, parse_arguments +from metriq_gym.benchmarks.bseq import BSEQData +from metriq_gym.benchmarks.quantum_volume import QuantumVolumeData from metriq_gym.job_manager import JobManager, MetriqGymJob from metriq_gym.schema_validator import load_and_validate from metriq_gym.job_type import JobType @@ -102,7 +104,7 @@ def poll_job(args: argparse.Namespace, job_manager: JobManager, is_upload: bool= logger.info("Job is not yet completed. Please try again later.") -def upload_job(args: argparse.Namespace, job_type: JobType, job_data: BenchmarkData, result_data: BenchmarkResult, platform: int): +def upload_job(args: argparse.Namespace, job_type: JobType, job_data: QuantumVolumeData | BSEQData, result_data: BenchmarkResult, platform: int): client = MetriqClient(os.environ.get("METRIQ_CLIENT_API_KEY")) task = 0 method = 0