forgejoadmin/cubescripts
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix typing of cube_pool.load_pools()

Browse source
This commit is contained in:
henriquenakashima 2021-02-22 21:27:29 -05:00
parent b57ddacefd
commit f8b5d804df
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
cube_pool.py
View file

@ -4,7 +4,7 @@ import csv
import itertools import itertools
import random import random
from collections import defaultdict from collections import defaultdict
from typing import Dict, List, Set from typing import Dict, Set
import cubecobra_csv import cubecobra_csv
@ -22,7 +22,7 @@ class CubePool:
return random.sample(wide_iterator, n) 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) pools = defaultdict(CubePool)
with open(csv_path) as f: with open(csv_path) as f:
reader = csv.reader(f) reader = csv.reader(f)