From f8b5d804dfc8f7411a44adf973b48ee241ba4aa1 Mon Sep 17 00:00:00 2001 From: henriquenakashima Date: Mon, 22 Feb 2021 21:27:29 -0500 Subject: [PATCH] Fix typing of cube_pool.load_pools() --- cube_pool.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cube_pool.py b/cube_pool.py index 348e170..44c4982 100644 --- a/cube_pool.py +++ b/cube_pool.py @@ -4,7 +4,7 @@ import csv import itertools import random from collections import defaultdict -from typing import Dict, List, Set +from typing import Dict, Set import cubecobra_csv @@ -22,7 +22,7 @@ class CubePool: return random.sample(wide_iterator, n) -def load_pools(csv_path: str, pool_tags: Set[str]) -> Dict[str, List[str]]: +def load_pools(csv_path: str, pool_tags: Set[str]) -> Dict[str, CubePool]: pools = defaultdict(CubePool) with open(csv_path) as f: reader = csv.reader(f)