Skip to content

Commit 15e11c1

Browse files
committed
add missing normalize config call to test
1 parent 12a6d4b commit 15e11c1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_networks/test_point_inference_network/test_point_inference_network.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
deserialize_keras_object as deserialize,
44
serialize_keras_object as serialize,
55
)
6-
from tests.utils import assert_layers_equal
6+
from tests.utils import assert_layers_equal, normalize_config
77
import pytest
88

99

@@ -72,7 +72,7 @@ def test_save_and_load_quantile(tmp_path, quantile_point_inference_network, rand
7272
loaded = keras.saving.load_model(tmp_path / "model.keras")
7373

7474
print(net.get_config())
75-
assert net.get_config() == loaded.get_config()
75+
assert normalize_config(net.get_config()) == normalize_config(loaded.get_config())
7676

7777
assert_layers_equal(net, loaded)
7878

0 commit comments

Comments
 (0)