Skip to content

Commit a9068a8

Browse files
committed
Remove bytes option for filename
1 parent 9ec1ccb commit a9068a8

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

axelrod/load_data_.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,7 @@
11
import pkg_resources
22
from typing import List, Dict, Tuple, Union
33

4-
filenameType = Union[str, bytes]
5-
6-
def load_file(filename: filenameType, directory: str) -> List[List[str]]:
4+
def load_file(filename: str, directory: str) -> List[List[str]]:
75
"""Loads a data file stored in the Axelrod library's data subdirectory,
86
likely for parameters for a strategy."""
97
path = '/'.join((directory, filename))
@@ -18,7 +16,7 @@ def load_file(filename: filenameType, directory: str) -> List[List[str]]:
1816
return rows
1917

2018

21-
def load_weights(filename: filenameType ="ann_weights.csv", directory: str ="data") -> Dict[str, Tuple[int, int, List[float]]]:
19+
def load_weights(filename: str ="ann_weights.csv", directory: str ="data") -> Dict[str, Tuple[int, int, List[float]]]:
2220
"""Load Neural Network Weights."""
2321
rows = load_file(filename, directory)
2422
d = dict()
@@ -31,7 +29,7 @@ def load_weights(filename: filenameType ="ann_weights.csv", directory: str ="dat
3129
return d
3230

3331

34-
def load_lookerup_tables(filename: filenameType ="lookup_tables.csv", directory: str ="data") -> Dict[Tuple[str, int, int, int], Tuple[str, str]]:
32+
def load_lookerup_tables(filename: str ="lookup_tables.csv", directory: str ="data") -> Dict[Tuple[str, int, int, int], Tuple[str, str]]:
3533
"""Load lookup tables."""
3634
rows = load_file(filename, directory)
3735
d = dict()
@@ -41,7 +39,7 @@ def load_lookerup_tables(filename: filenameType ="lookup_tables.csv", directory:
4139
return d
4240

4341

44-
def load_pso_tables(filename: filenameType ="pso_gambler.csv", directory: str ="data") -> Dict[Tuple[str, int, int, int], List[float]]:
42+
def load_pso_tables(filename: str ="pso_gambler.csv", directory: str ="data") -> Dict[Tuple[str, int, int, int], List[float]]:
4543
"""Load lookup tables."""
4644
rows = load_file(filename, directory)
4745
d = dict()

0 commit comments

Comments
 (0)