Skip to content

Commit 4a45b8c

Browse files
committed
Revert move to TF 2.10-rc1
This reverts commits 6ec41b9 and 1804d49. A runtime error is introduced, needs more investigation. See https://github.com/google/ml-compiler-opt/runs/7932290113?check_suite_focus=true
1 parent 1804d49 commit 4a45b8c

File tree

2 files changed

+9
-8
lines changed

2 files changed

+9
-8
lines changed

compiler_opt/rl/constant_value_network.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,8 +45,9 @@ def __init__(self, input_tensor_spec, constant_output_val=0, name=None):
4545

4646
self._constant_output_val = constant_output_val
4747

48-
def call(self, inputs, step_type=None, network_state=(), training=False):
48+
def call(self, observation, step_type=None, network_state=(), training=False):
4949
_ = (step_type, training)
50-
shape = nest_utils.get_outer_array_shape(inputs, self._input_tensor_spec)
50+
shape = nest_utils.get_outer_array_shape(observation,
51+
self._input_tensor_spec)
5152
return tf.constant(
5253
self._constant_output_val, tf.float32, shape=shape), network_state

requirements.txt

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
absl-py==1.0.0
1+
absl-py==0.11.0
22
astunparse==1.6.3
33
cachetools==4.2.2
44
certifi==2021.5.30
@@ -18,9 +18,9 @@ grpcio==1.39.0
1818
gym==0.19.0
1919
h5py==3.6.0
2020
idna==3.2
21-
keras==2.10-rc1
21+
keras==2.7.0
2222
keras-preprocessing==1.1.2
23-
libclang==13.0.0
23+
libclang==12.0.0
2424
markdown==3.3.4
2525
numpy==1.22.1
2626
oauthlib==3.1.1
@@ -37,11 +37,11 @@ rsa==4.7.2
3737
scipy==1.7.1
3838
setuptools==57.4.0
3939
six==1.16.0
40-
tensorboard==2.9
40+
tensorboard==2.7.0
4141
tensorboard-data-server==0.6.1
4242
tensorboard-plugin-wit==1.8.0
43-
tensorflow==2.10-RC1
44-
tensorflow-estimator==2.10-rc0
43+
tensorflow==2.7.0
44+
tensorflow-estimator==2.7.0
4545
tensorflow-io-gcs-filesystem==0.23.1
4646
tensorflow-probability==0.15.0
4747
termcolor==1.1.0

0 commit comments

Comments
 (0)