7ab8bc6a70982aa79c36c20678c98becb7f291b1,deepctr/estimator/utils.py,,get_GraphKeys,#,179

Before Change

def get_GraphKeys():
    if tf.__version__ < "2.0.0":
        return tf.GraphKeys
    else:
        return tf.compat.v1.GraphKeys


def get_losses():
    if tf.__version__ < "2.0.0":
        return tf.losses
    else:

After Change

def get_GraphKeys():
    try:
        return tf.GraphKeys
    except AttributeError:
        return tf.compat.v1.GraphKeys


def get_losses():
    try:
        return tf.compat.v1.losses
    except AttributeError:

Instances