From ea41f5e1c1106a00a55a7874761868032c2a32be Mon Sep 17 00:00:00 2001 From: George Hotz Date: Sun, 25 Oct 2020 16:40:37 -0700 Subject: [PATCH] seems more generic --- test/test_mnist.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_mnist.py b/test/test_mnist.py index 4c85e3426d..1affa06d52 100644 --- a/test/test_mnist.py +++ b/test/test_mnist.py @@ -39,7 +39,7 @@ class TinyConvNet: def train(model, optim, steps, BS=128): losses, accuracies = [], [] - for i in (t := trange(steps, disable=os.getenv('GITHUB_ACTIONS') is not None)): + for i in (t := trange(steps, disable=os.getenv('CI') is not None)): samp = np.random.randint(0, X_train.shape[0], size=(BS)) x = Tensor(X_train[samp].reshape((-1, 28*28)).astype(np.float32))