Skip to content

Commit b789727

Browse files
committed
Correctly load datasets from egg distribution.
The datasets module failed to load the datasets when reading from an egg. Now fixed.
1 parent e61e15e commit b789727

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

gmaps/datasets/datasets.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -22,22 +22,21 @@
2222
METADATA_FNAME = "metadata.yaml"
2323

2424
def _load_metadata():
25-
with pkg_resources.resource_stream(__name__, METADATA_FNAME) as f:
26-
datasets = yaml.load(f)
25+
f = pkg_resources.resource_stream(__name__, METADATA_FNAME)
26+
datasets = yaml.load(f)
27+
f.close()
2728
return datasets
2829

2930
def list_datasets():
3031
metadata = _load_metadata()
3132
return metadata.keys()
3233

33-
metadata = _load_metadata()
34-
return metadata[dataset_name]
35-
3634
def load_dataset(dataset_name):
3735
metadata = _load_metadata()
3836
fname = metadata[dataset_name]["data_file"]
3937
fpath = os.path.join(DATA_DIR, fname)
40-
with pkg_resources.resource_stream(__name__, fpath) as f:
41-
data = np.genfromtxt(f, delimiter=",", names=True)
38+
f = pkg_resources.resource_stream(__name__, fpath)
39+
data = np.genfromtxt(f, delimiter=",", names=True)
40+
f.close()
4241
return data
4342

0 commit comments

Comments
 (0)