fastmachinelearning / hls4ml

Machine learning on FPGAs using HLS
https://fastmachinelearning.org/hls4ml
Apache License 2.0
1.24k stars 401 forks source link

Fix extension test for Keras v3 #1009

Closed vloncar closed 4 months ago

vloncar commented 5 months ago

Description

Newer versions of TF/Keras change the way serialization is handled, so one needs to define a get_config() to ensure the serialized json contains the information about the layer.

Type of change

Fixes a bug observed in new testing environment, but is benign for older setups. I updated the docs to ensure this info is not hidden away in some test no user ever sees.

Tests

I extended the test in test_extensions.py.

Checklist