Skip to content

Commit d154c5d

Browse files
authored
Remove Version Check for tf.estimator (#1460)
Signed-off-by: zehao-intel <zehao.huang@intel.com>
1 parent 4ca8d1f commit d154c5d

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

neural_compressor/model/tensorflow_model.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -81,8 +81,7 @@ def get_model_type(model):
8181
return "graph"
8282
elif isinstance(model, tf.compat.v1.GraphDef):
8383
return "graph_def"
84-
# tf.estimator is removed after tf2.14.0
85-
elif version1_lt_version2(tf.__version__, "2.14.0") and isinstance(model, tf.compat.v1.estimator.Estimator):
84+
elif isinstance(model, tf.compat.v1.estimator.Estimator):
8685
return "estimator"
8786
elif isinstance(model, str):
8887
model = os.path.abspath(os.path.expanduser(model))

0 commit comments

Comments
 (0)