Undefined name: import os for lines 13 and 14

2020-01-16 13:37:14 +01:00 · 2020-01-16 13:37:14 +01:00 · 0922f9916d
commit 0922f9916d
parent b299945b23
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)