Skip to content

Commit

Permalink
Improve the tensorflow frontend _test_spop_resource_variables to supp…
Browse files Browse the repository at this point in the history
…ort tensoflow 2.6 (#9978)

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
  • Loading branch information
ophirfrish committed Jan 26, 2022
1 parent 095b639 commit cc8a7a2
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion tests/python/frontend/tensorflow/test_forward.py
Original file line number Diff line number Diff line change
Expand Up @@ -5391,7 +5391,12 @@ def resourceVariablesTest(x, y):
def test_forward_spop():
_test_spop_stateful()
_test_spop_device_assignment()
_test_spop_resource_variables()
# tensorflow version upgrade support
# This test is expected to fail in TF version >= 2.6
# as the generated graph will be considered frozen, hence
# not passing the criteria for the test below.
if tf.__version__ < LooseVersion("2.6.1"):
_test_spop_resource_variables()

# Placeholder test cases
_test_spop_placeholder_without_shape_info()
Expand Down

0 comments on commit cc8a7a2

Please sign in to comment.