From de5caa7a0131ffcd13959c919c59d296dd72d134 Mon Sep 17 00:00:00 2001
From: Kevin Escobar Rodriguez <kevin.rodriguez@neuralmagic.com>
Date: Wed, 3 Feb 2021 12:34:49 -0500
Subject: [PATCH] removed faulty tests in resnet

---
 tests/sparsezoo/models/classification/test_resnet.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/tests/sparsezoo/models/classification/test_resnet.py b/tests/sparsezoo/models/classification/test_resnet.py
index 45dad226..1c6e7f29 100644
--- a/tests/sparsezoo/models/classification/test_resnet.py
+++ b/tests/sparsezoo/models/classification/test_resnet.py
@@ -139,10 +139,10 @@ def test_resnet_50(
             True,
             "pytorch",
             "sparseml",
-            "imagenet",
+            "imagenet-augmented",
             None,
             "pruned_quant",
-            "conservative",
+            "aggressive",
             None,
         ),
         (True, "tensorflow_v1", "sparseml", "imagenet", None, "base", "none", None),
@@ -157,7 +157,6 @@ def test_resnet_50(
             "conservative",
             None,
         ),
-        (True, "pytorch", "sparseml", "imagenette", None, "pruned", "moderate", None),
         (
             True,
             "pytorch",