Merge pull request #49 from cclauss/patch-1

Undefined name: import os for lines 13 and 14 /lib/datasets/test_utils.py
2020-01-17 10:22:01 +11:00
parent b299945b23 0922f9916d
commit 3ad965d510
1 changed files with 3 additions and 0 deletions
--- a/lib/datasets/test_utils.py
+++ b/lib/datasets/test_utils.py
@@ -1,6 +1,9 @@
 ##################################################
 # Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2019 #
 ##################################################
+import os
+
+
 def test_imagenet_data(imagenet):
  total_length = len(imagenet)
  assert total_length == 1281166 or total_length == 50000, 'The length of ImageNet is wrong : {}'.format(total_length)