From e242bbce65cd498149ff143b2081be19f30e88f3 Mon Sep 17 00:00:00 2001 From: Florian Fischer Date: Sun, 19 Jul 2020 15:49:33 +0200 Subject: [benchmark] use AlloactorCollection type --- allocbench/benchmark.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/allocbench/benchmark.py b/allocbench/benchmark.py index 53430cb..bd68db1 100644 --- a/allocbench/benchmark.py +++ b/allocbench/benchmark.py @@ -34,6 +34,7 @@ from typing import Any, Dict, Iterable, List, Optional, Union, cast import numpy as np +from allocbench.allocator import AllocatorCollection from allocbench.directories import (get_allocbench_benchmark_src_dir, get_current_result_dir, get_allocbench_benchmark_build_dir, @@ -546,7 +547,7 @@ class Benchmark: for k in valid_result } - def run(self, allocators: Dict[str, Dict[str, str]], runs=3): + def run(self, allocators: AllocatorCollection, runs=3): """generic run implementation""" self.results["allocators"] = copy.deepcopy(allocators) -- cgit v1.2.3