From 075653e6ddde7db380acff986771bfcacbf91320 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 7 Aug 2019 12:01:18 -0400
Subject: [PATCH] moved badge image

---
 .codecov.yml                                | 4 ++++
 pytorch_lightning/testing/lm_test_module.py | 6 +++---
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/.codecov.yml b/.codecov.yml
index f10c1873..40560393 100644
--- a/.codecov.yml
+++ b/.codecov.yml
@@ -40,3 +40,7 @@ comment:
   require_changes: false
   behavior: default  # update if exists else create new
   # branches: *
+
+ignore:
+  - "pytorch_lightning/utilities/arg_parse.py"
+  -
diff --git a/pytorch_lightning/testing/lm_test_module.py b/pytorch_lightning/testing/lm_test_module.py
index 695b495f..453dc145 100644
--- a/pytorch_lightning/testing/lm_test_module.py
+++ b/pytorch_lightning/testing/lm_test_module.py
@@ -135,16 +135,16 @@ class LightningTestModel(LightningModule):
             val_acc = val_acc.unsqueeze(0)
 
         # alternate possible outputs to test
-        if self.trainer.batch_nb % 1 == 0:
+        if batch_i % 1 == 0:
             output = OrderedDict({
                 'val_loss': loss_val,
                 'val_acc': val_acc,
             })
             return output
-        if self.trainer.batch_nb % 2 == 0:
+        if batch_i % 2 == 0:
             return val_acc
 
-        if self.trainer.batch_nb % 3 == 0:
+        if batch_i % 3 == 0:
             output = OrderedDict({
                 'val_loss': loss_val,
                 'val_acc': val_acc,