Skip to content

Commit 26ac44e

Browse files
committed
MAINT refactor nested loop
1 parent 2bb3bf2 commit 26ac44e

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

openml/datasets/functions.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -193,16 +193,16 @@ def check_datasets_active(dids):
193193
dids = sorted(dids)
194194
active = {}
195195

196-
for did in dids:
197-
for dataset in dataset_list:
198-
if did == int(dataset['did']):
199-
active[did] = dataset['status'] == 'active'
196+
for dataset in dataset_list:
197+
active[dataset['did']] = dataset['status'] == 'active'
200198

201199
for did in dids:
202200
if did not in active:
203201
raise ValueError('Could not find dataset %d in OpenML dataset list.'
204202
% did)
205203

204+
active = {did: active[did] for did in dids}
205+
206206
return active
207207

208208

0 commit comments

Comments
 (0)