From f06662d39d200ca6cc6903339e55bbb4e10d115e Mon Sep 17 00:00:00 2001
From: Guy Jacob <guy.jacob@intel.com>
Date: Tue, 14 May 2019 17:14:52 +0300
Subject: [PATCH] Fix failing test after last post-train quant change
---
tests/test_post_train_quant.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/tests/test_post_train_quant.py b/tests/test_post_train_quant.py
index 5d77067..4522e01 100644
--- a/tests/test_post_train_quant.py
+++ b/tests/test_post_train_quant.py
@@ -241,7 +241,7 @@ def test_concat_layer_wrapper(inputs, concat_stats, mode, clip_acts, expected_ou
layer = distiller.modules.Concat(dim=1)
- with pytest.raises(ValueError):
+ with pytest.raises(NotImplementedError):
# Check exception on no stats
RangeLinearQuantConcatWrapper(layer, 8, mode, clip_acts, activation_stats=None)
@@ -292,7 +292,7 @@ def test_eltwise_mult_layer_wrapper(inputs, eltwise_mult_stats, mode, clip_acts,
layer = distiller.modules.EltwiseMult()
- with pytest.raises(ValueError):
+ with pytest.raises(NotImplementedError):
# Check exception on no stats
RangeLinearQuantEltwiseMultWrapper(layer, 8, mode, clip_acts, activation_stats=None)
@@ -343,7 +343,7 @@ def test_eltwise_add_layer_wrapper(inputs, eltwise_add_stats, mode, clip_acts, e
layer = distiller.modules.EltwiseAdd()
- with pytest.raises(ValueError):
+ with pytest.raises(NotImplementedError):
# Check exception on no stats
RangeLinearQuantEltwiseAddWrapper(layer, 8, mode, clip_acts, activation_stats=None)
--
GitLab