Use proper namespace for constants.

lmjohns3 · Jun 7, 2015 · 447f196 · 447f196
1 parent 4790d83
commit 447f196
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/util.py b/test/util.py
@@ -10,5 +10,5 @@ class MNIST(object):
     def setUp(self):
         # we just create some random "mnist digit" data of the right shape.
         np.random.seed(3)
-        self.images = np.random.randn(NUM_DIGITS, DIGIT_SIZE).astype('f')
-        self.labels = np.random.randint(0, 10, NUM_DIGITS).astype('i')
+        self.images = np.random.randn(MNIST.NUM_DIGITS, MNIST.DIGIT_SIZE).astype('f')
+        self.labels = np.random.randint(0, 10, MNIST.NUM_DIGITS).astype('i')