diff --git a/test/pytest/test_cnn_mnist_qkeras.py b/test/pytest/test_cnn_mnist_qkeras.py index 6e88756332..0edfd16f9f 100644 --- a/test/pytest/test_cnn_mnist_qkeras.py +++ b/test/pytest/test_cnn_mnist_qkeras.py @@ -45,7 +45,7 @@ def mnist_model(): def hls_model(mnist_model, backend, io_type, strategy): keras_model = mnist_model hls_config = hls4ml.utils.config_from_keras_model(keras_model, granularity='name') - hls_config['Model']['Strategy'] = 'Resource' + hls_config['Model']['Strategy'] = strategy hls_config['LayerName']['softmax']['Strategy'] = 'Stable' output_dir = str(test_root_path / 'hls4mlprj_cnn_mnist_qkeras_{}_{}_{}'.format(backend, io_type, strategy))