Since tf 2.17 version, keras._tf_keras.keras.backend.reset_uids does not exist anymore which breaks CI.