tensorflow get_config error

[dyuliu]

• MLPrimitives version: 0.3.0
• Python version: 3.6

Description

The current version of MLPrimitives will automatically install tensorflow 2.3.4.

This version will encounter the following issue:

/usr/local/lib/python3.6/site-packages/keras/backend.py in <module>
     34 from tensorflow.core.protobuf import config_pb2
     35 from tensorflow.python.eager import context
---> 36 from tensorflow.python.eager.context import get_config
     37 from tensorflow.python.framework import config
     38 from keras import backend_config

ImportError: cannot import name 'get_config'

Because of the piece of code from mlprimitives/adapters/keras.py

import logging
import tempfile

import keras        # this is the line causing error
import numpy as np

Solution

Simply replace

import keras

as

from tensorflow import keras