import tensorflow
tf = tensorflow.compat.v1
tf.disable_v2_behavior()
TF_VERSION = tf.__version__


def get_tf_version(version):
    if version is not None:
        assert isinstance(version, str)
        return version
    else:
        return TF_VERSION

0eb53027388d37f55fb520f618bbfc73cd52251b
<mask token>


def get_tf_version(version):
    if version is not None:
        assert isinstance(version, str)
        return version
    else:
        return TF_VERSION

<mask token>
tf.disable_v2_behavior()
<mask token>


def get_tf_version(version):
    if version is not None:
        assert isinstance(version, str)
        return version
    else:
        return TF_VERSION

<mask token>
tf = tensorflow.compat.v1
tf.disable_v2_behavior()
TF_VERSION = tf.__version__


def get_tf_version(version):
    if version is not None:
        assert isinstance(version, str)
        return version
    else:
        return TF_VERSION

import tensorflow
tf = tensorflow.compat.v1
tf.disable_v2_behavior()
TF_VERSION = tf.__version__


def get_tf_version(version):
    if version is not None:
        assert isinstance(version, str)
        return version
    else:
        return TF_VERSION
