diff --git a/examples/example_compiler_gym_service/__init__.py b/examples/example_compiler_gym_service/__init__.py index 3349b9867..151cbb2fa 100644 --- a/examples/example_compiler_gym_service/__init__.py +++ b/examples/example_compiler_gym_service/__init__.py @@ -72,7 +72,7 @@ def __init__(self, *args, **kwargs): def benchmark_uris(self) -> Iterable[str]: yield from (f"benchmark://example-v0{k}" for k in self._benchmarks.keys()) - def benchmark_from_parsed_uris(self, uri: BenchmarkUri) -> Benchmark: + def benchmark_from_parsed_uri(self, uri: BenchmarkUri) -> Benchmark: if uri.path in self._benchmarks: return self._benchmarks[uri] else: