From 0922f9916d58136843a2303b5f16ff9d2dc28a6f Mon Sep 17 00:00:00 2001
From: Christian Clauss <cclauss@me.com>
Date: Thu, 16 Jan 2020 13:37:14 +0100
Subject: [PATCH] Undefined name: import os for lines 13 and 14
---
lib/datasets/test_utils.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/lib/datasets/test_utils.py b/lib/datasets/test_utils.py
index ca30512..2245f87 100644
--- 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)