Skip to content

Commit 092b235

Browse files
committed
Improve the tensorflow frontend _test_spop_resource_variables to support tensoflow 2.6
On tensorflow 2.4 the test is expected to fail as the generated graph is not forzen. On tensorflow 2.6 the generated graph is identified as frozen, therefore the test is not needed
1 parent 884fee4 commit 092b235

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tests/python/frontend/tensorflow/test_forward.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5445,7 +5445,12 @@ def resourceVariablesTest(x, y):
54455445
def test_forward_spop():
54465446
_test_spop_stateful()
54475447
_test_spop_device_assignment()
5448-
_test_spop_resource_variables()
5448+
# tensorflow version upgrade support
5449+
# This test is expected to fail in TF version >= 2.6
5450+
# as the generated graph will be considered frozen, hence
5451+
# not passing the criteria for the test below.
5452+
if tf.__version__ < LooseVersion("2.6.1"):
5453+
_test_spop_resource_variables()
54495454

54505455
# Placeholder test cases
54515456
_test_spop_placeholder_without_shape_info()

0 commit comments

Comments
 (0)