Skip to content

Commit 872dedb

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 c3ace20 commit 872dedb

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
@@ -5435,7 +5435,12 @@ def resourceVariablesTest(x, y):
54355435
def test_forward_spop():
54365436
_test_spop_stateful()
54375437
_test_spop_device_assignment()
5438-
_test_spop_resource_variables()
5438+
# tensorflow version upgrade support
5439+
# This test is expected to fail in TF version >= 2.6
5440+
# as the generated graph will be considered frozen, hence
5441+
# not passing the criteria for the test below.
5442+
if tf.__version__ < LooseVersion("2.6.1"):
5443+
_test_spop_resource_variables()
54395444

54405445
# Placeholder test cases
54415446
_test_spop_placeholder_without_shape_info()

0 commit comments

Comments
 (0)