From a7809f39d2eca4b02de957bc1d5db0fc761199ab Mon Sep 17 00:00:00 2001 From: Aiden Grossman Date: Tue, 31 Dec 2024 00:33:30 +0000 Subject: [PATCH] =?UTF-8?q?[=F0=9D=98=80=F0=9D=97=BD=F0=9D=97=BF]=20change?= =?UTF-8?q?s=20introduced=20through=20rebase?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Created using spr 1.3.4 [skip ci] --- gematria/datasets/pipelines/benchmark_cpu_scheduler.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/gematria/datasets/pipelines/benchmark_cpu_scheduler.py b/gematria/datasets/pipelines/benchmark_cpu_scheduler.py index 665224fd..87a4ef85 100644 --- a/gematria/datasets/pipelines/benchmark_cpu_scheduler.py +++ b/gematria/datasets/pipelines/benchmark_cpu_scheduler.py @@ -13,7 +13,7 @@ # limitations under the License. import abc -import typing_extensions +from typing_extensions import override from collections.abc import Iterable import os import re @@ -63,11 +63,11 @@ class NoSchedulingBenchmarkScheduler(BenchmarkScheduler): performs no verification. """ - @typing_extensions.override + @override def setup_and_get_benchmark_core(self) -> int | None: return None - @typing_extensions.override + @override def verify(self): pass @@ -122,7 +122,7 @@ def _get_aux_core_and_hyperthread_pair( 'Expected a pair of neighboring hyperthreads in the CPU mask.' ) - @typing_extensions.override + @override def setup_and_get_benchmark_core(self) -> int | None: cpu_mask = os.sched_getaffinity(0) @@ -137,7 +137,7 @@ def setup_and_get_benchmark_core(self) -> int | None: return hyperthread_pair[0] - @typing_extensions.override + @override def verify(self): cpu_mask = list(os.sched_getaffinity(0)) if self._cpu_mask != cpu_mask: