diff --git a/n3fit/src/n3fit/tests/test_multidense.py b/n3fit/src/n3fit/tests/test_multidense.py index 4ffbfc4a53..b912c0bd82 100644 --- a/n3fit/src/n3fit/tests/test_multidense.py +++ b/n3fit/src/n3fit/tests/test_multidense.py @@ -4,7 +4,7 @@ from tensorflow.keras.initializers import GlorotUniform from tensorflow.keras.layers import Dense -from n3fit.backends.keras_backend.multi_dense import MultiDense, MultiDropout +from n3fit.backends.keras_backend.multi_dense import MultiDense from n3fit.model_gen import generate_nn