Fix keras 2.4 autolog#3571 (#3611)
Fix keras 2.4 autolog#3571 (#3611)

* Fix bug - when running Keras 2.4.x

Signed-off-by: sephi berry <39e52bac36ba881dc78ae445bc3b3c77b70bce2f@gmail.com>

* fix black

Signed-off-by: sephi berry <39e52bac36ba881dc78ae445bc3b3c77b70bce2f@gmail.com>
mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow