Skip to content

Commit 44105b7

Browse files
committed
Remove config comparison for heads
1 parent 23790c6 commit 44105b7

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

tests/test_networks/test_point_inference_network/test_point_inference_network.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -88,8 +88,5 @@ def test_save_and_load_quantile(tmp_path, quantile_point_inference_network, rand
8888
assert keras.ops.all(keras.ops.isclose(net_head.layers[-1].q, loaded_head.layers[-1].q))
8989
assert keras.ops.all(net_head.layers[-1].anchor_index == loaded_head.layers[-1].anchor_index)
9090

91-
print(net_head.get_config())
92-
assert net_head.get_config() == loaded_head.get_config()
93-
9491
print(f"Asserting original and serialized and deserialized at heads[{score_key}][{head_key}] to be equal")
9592
assert_layers_equal(net_head, loaded_head)

0 commit comments

Comments
 (0)