test batch=3

fastmachinelearning · Oct 12, 2021 · 81d6b42 · 81d6b42
1 parent 13bcce8
commit 81d6b42
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/pytest/test_garnet.py b/test/pytest/test_garnet.py
@@ -33,7 +33,7 @@ def garnet_models():
     return model, hls_model
 
 
-@pytest.mark.parametrize('batch', [1]) # [1, 3] when https://github.com/fastmachinelearning/hls4ml/pull/414 is merged
+@pytest.mark.parametrize('batch', [1, 3])
 def test_accuracy(garnet_models, batch):
     model, hls_model = garnet_models
     x = [np.random.rand(batch, vmax, feat), np.random.randint(0, vmax, size=(batch, 1))]