Skip to content

Commit 956808a

Browse files
committed
fix dsp_aware_pruning test import path
1 parent 6791412 commit 956808a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/pytest/test_optimization/test_attributes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
from tensorflow.keras.layers import Conv2D, Dense, Flatten, ReLU
22
from tensorflow.keras.models import Sequential
33

4-
from hls4ml.optimization import get_attributes_from_keras_model_and_hls4ml_config
4+
from hls4ml.optimization.dsp_aware_pruning import get_attributes_from_keras_model_and_hls4ml_config
55
from hls4ml.utils.config import config_from_keras_model
66

77

0 commit comments

Comments
 (0)