Skip to content

Commit 500284d

Browse files
authored
[TF FE][Keras] Use latest Keras for TF FE validation and adopt tests (#26912)
**Details:** New Keras Dot operation does not support integer operands **Ticket:** TBD Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>
1 parent 890f2e1 commit 500284d

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

tests/layer_tests/tensorflow2_keras_tests/test_tf2_keras_dot.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -44,9 +44,7 @@ def create_keras_dot_net(self, input_shapes, axes, input_type, normalize):
4444
([[5, 1, 5, 2], [5, 1, 2, 5]], (2, 3)),
4545
([[5, 1, 102, 2], [5, 1, 4, 102]], (-2, -1))
4646
])
47-
@pytest.mark.parametrize('input_type', [np.float32, np.float64,
48-
np.int8, np.int16, np.int32, np.int64,
49-
np.uint8, np.uint16, np.uint32, np.uint64])
47+
@pytest.mark.parametrize('input_type', [np.float32, np.float64])
5048
@pytest.mark.parametrize('normalize', [True, False])
5149
@pytest.mark.nightly
5250
@pytest.mark.precommit

tests/requirements_tensorflow

+2
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,8 @@ pytest==7.0.1
44
pytest-xdist[psutil]==3.6.1
55
pytest-html==4.1.1
66
transformers==4.45.1
7+
# install exact keras version since tensorflow depends and has no upper bound for it
8+
keras==3.6.0
79
tensorflow==2.17.0; platform_system != "Darwin" or platform_machine != "x86_64"
810
tensorflow==2.16.2; platform_system == "Darwin" and platform_machine == "x86_64"
911
# tensorflow-text is not available for both Windows and ARM platforms

0 commit comments

Comments
 (0)