restored_model = build_model()
restored_model.load_weights(weights_path)
prediction_after = np.asarray(
    restored_model(sample, training=False)
)
np.testing.assert_allclose(
    prediction_before,
    prediction_after,
    atol=1e-7,
)

print(f"weights file: {weights_path}")
print(
    "prediction before save: "
    f"{prediction_before[0, 0]:.6f}"
)
print(
    "prediction after load: "
    f"{prediction_after[0, 0]:.6f}"
)
print("predictions match: True")