Skip to content

Commit 2a41215

Browse files
authored
Merge pull request #160 from rtg0795/r0.41.0
Fix tfma.__version__.
2 parents 4a20f57 + 96b5547 commit 2a41215

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

tensorflow_model_analysis/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,9 @@
103103
from tensorflow_model_analysis.types import TensorValue
104104
from tensorflow_model_analysis.types import VarLenTensorValue
105105

106+
# Import VERSION as __version__ for compatibility with other TFX components.
107+
from tensorflow_model_analysis.version import VERSION as __version__
108+
106109
except ImportError as err:
107110
import sys
108111
sys.stderr.write('Error importing: {}'.format(err))

tensorflow_model_analysis/sdk.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,5 @@
8080
from tensorflow_model_analysis.proto.config_pb2 import SlicingSpec
8181
# pylint: enable=g-importing-member
8282

83-
# Import VERSION as __version__ for compatibility with other TFX components.
84-
from tensorflow_model_analysis.version import VERSION as __version__
8583
# Import VERSION as VERSION_STRING for backwards compatibility.
86-
from tensorflow_model_analysis.version import VERSION as VERSION_STRING # pylint: disable=reimported
84+
from tensorflow_model_analysis.version import VERSION as VERSION_STRING

0 commit comments

Comments
 (0)