Skip to content

Commit

Permalink
[TF FE][Keras] Use latest Keras for TF FE validation and adopt tests (o…
Browse files Browse the repository at this point in the history
…penvinotoolkit#26912)

**Details:** New Keras Dot operation does not support integer operands

**Ticket:** TBD

Signed-off-by: Kazantsev, Roman <[email protected]>
  • Loading branch information
rkazants authored Oct 7, 2024
1 parent 890f2e1 commit 500284d
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -44,9 +44,7 @@ def create_keras_dot_net(self, input_shapes, axes, input_type, normalize):
([[5, 1, 5, 2], [5, 1, 2, 5]], (2, 3)),
([[5, 1, 102, 2], [5, 1, 4, 102]], (-2, -1))
])
@pytest.mark.parametrize('input_type', [np.float32, np.float64,
np.int8, np.int16, np.int32, np.int64,
np.uint8, np.uint16, np.uint32, np.uint64])
@pytest.mark.parametrize('input_type', [np.float32, np.float64])
@pytest.mark.parametrize('normalize', [True, False])
@pytest.mark.nightly
@pytest.mark.precommit
Expand Down
2 changes: 2 additions & 0 deletions tests/requirements_tensorflow
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@ pytest==7.0.1
pytest-xdist[psutil]==3.6.1
pytest-html==4.1.1
transformers==4.45.1
# install exact keras version since tensorflow depends and has no upper bound for it
keras==3.6.0
tensorflow==2.17.0; platform_system != "Darwin" or platform_machine != "x86_64"
tensorflow==2.16.2; platform_system == "Darwin" and platform_machine == "x86_64"
# tensorflow-text is not available for both Windows and ARM platforms
Expand Down

0 comments on commit 500284d

Please sign in to comment.