From d2c0435aaa359703582651ea51d9ddcdb16e73ef Mon Sep 17 00:00:00 2001 From: Adrian RC Date: Tue, 21 May 2024 00:01:18 +0000 Subject: [PATCH] Updates diffs in tests to unpin keras version for CI runs. --- .github/workflows/ci.yml | 2 +- tensorboard/plugins/graph/keras_util_test.py | 114 ------------------- 2 files changed, 1 insertion(+), 115 deletions(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 60fd4d9513..3aaf37a5b5 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -76,7 +76,7 @@ jobs: - name: 'Install TensorFlow' run: | python -m pip install -U pip - pip install "${TENSORFLOW_VERSION}" keras-nightly==3.3.3.dev2024051503 + pip install "${TENSORFLOW_VERSION}" if: matrix.tf_version_id != 'notf' - name: 'Install Python dependencies' run: | diff --git a/tensorboard/plugins/graph/keras_util_test.py b/tensorboard/plugins/graph/keras_util_test.py index 6c4a198e23..66bf8494ca 100644 --- a/tensorboard/plugins/graph/keras_util_test.py +++ b/tensorboard/plugins/graph/keras_util_test.py @@ -148,12 +148,6 @@ def test_keras_model_to_graph_def_functional_model(self): node { name: "model/dense" input: "model/functional_input" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } attr { key: "keras_class" value { @@ -164,12 +158,6 @@ def test_keras_model_to_graph_def_functional_model(self): node { name: "model/dense_1" input: "model/dense" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } attr { key: "keras_class" value { @@ -180,12 +168,6 @@ def test_keras_model_to_graph_def_functional_model(self): node { name: "model/dense_2" input: "model/dense_1" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } attr { key: "keras_class" value { @@ -224,12 +206,6 @@ def test_keras_model_to_graph_def_functional_model_with_cycle(self): node { name: "model/dense" input: "model/cycle_input" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } attr { key: "keras_class" value { @@ -241,12 +217,6 @@ def test_keras_model_to_graph_def_functional_model_with_cycle(self): name: "model/dense_1" input: "model/dense" input: "model/dense_2" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } attr { key: "keras_class" value { @@ -257,12 +227,6 @@ def test_keras_model_to_graph_def_functional_model_with_cycle(self): node { name: "model/dense_2" input: "model/dense_1" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } attr { key: "keras_class" value { @@ -301,12 +265,6 @@ def test_keras_model_to_graph_def_lstm_model(self): node { name: "model/simple_rnn" input: "model/lstm_input" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } attr { key: "keras_class" value { @@ -497,12 +455,6 @@ def test_keras_model_to_graph_def_functional_multi_inputs(self): node { name: "model/embedding" input: "model/main_input" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } attr { key: "keras_class" value { @@ -513,12 +465,6 @@ def test_keras_model_to_graph_def_functional_multi_inputs(self): node { name: "model/simple_rnn" input: "model/embedding" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } attr { key: "keras_class" value { @@ -545,12 +491,6 @@ def test_keras_model_to_graph_def_functional_multi_inputs(self): name: "model/concatenate" input: "model/simple_rnn" input: "model/aux_input" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } attr { key: "keras_class" value { @@ -561,12 +501,6 @@ def test_keras_model_to_graph_def_functional_multi_inputs(self): node { name: "model/dense" input: "model/concatenate" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } attr { key: "keras_class" value { @@ -577,12 +511,6 @@ def test_keras_model_to_graph_def_functional_multi_inputs(self): node { name: "model/main_output" input: "model/dense" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } attr { key: "keras_class" value { @@ -593,12 +521,6 @@ def test_keras_model_to_graph_def_functional_multi_inputs(self): node { name: "model/aux_output" input: "model/simple_rnn" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } attr { key: "keras_class" value { @@ -700,12 +622,6 @@ def test_keras_model_to_graph_def_functional_model_as_layer(self): node { name: "model_1/model/dense" input: "model_1/model/sub_func_input_1" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } attr { key: "keras_class" value { @@ -716,12 +632,6 @@ def test_keras_model_to_graph_def_functional_model_as_layer(self): node { name: "model_1/model/dense_1" input: "model_1/model/sub_func_input_2" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } attr { key: "keras_class" value { @@ -733,12 +643,6 @@ def test_keras_model_to_graph_def_functional_model_as_layer(self): name: "model_1/concatenate" input: "model_1/model/dense" input: "model_1/model/dense_1" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } attr { key: "keras_class" value { @@ -749,12 +653,6 @@ def test_keras_model_to_graph_def_functional_model_as_layer(self): node { name: "model_1/dense_2" input: "model_1/concatenate" - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } attr { key: "keras_class" value { @@ -1010,12 +908,6 @@ def test_keras_model_to_graph_def_functional_multiple_inbound_nodes_from_same_no s: "_DoublingLayer" } } - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } } node { name: "functional_1/add" @@ -1027,12 +919,6 @@ def test_keras_model_to_graph_def_functional_multiple_inbound_nodes_from_same_no s: "Add" } } - attr { - key: "dtype" - value { - type: DT_FLOAT - } - } } """ inputs = tf.keras.Input(shape=(2,))