Merge pull request #49 from cclauss/patch-1

Undefined name: import os for lines 13 and 14 /lib/datasets/test_utils.py
This commit is contained in:
D-X-Y 2020-01-17 10:22:01 +11:00 committed by GitHub
commit 3ad965d510
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,6 +1,9 @@
################################################## ##################################################
# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2019 # # Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2019 #
################################################## ##################################################
import os
def test_imagenet_data(imagenet): def test_imagenet_data(imagenet):
total_length = len(imagenet) total_length = len(imagenet)
assert total_length == 1281166 or total_length == 50000, 'The length of ImageNet is wrong : {}'.format(total_length) assert total_length == 1281166 or total_length == 50000, 'The length of ImageNet is wrong : {}'.format(total_length)