Skip to content

Commit 1804d49

Browse files
committed
Fix arguments-renamed warning introduced by TF 2.10
1 parent 6ec41b9 commit 1804d49

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

compiler_opt/rl/constant_value_network.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -45,9 +45,8 @@ 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, observation, step_type=None, network_state=(), training=False):
48+
def call(self, inputs, step_type=None, network_state=(), training=False):
4949
_ = (step_type, training)
50-
shape = nest_utils.get_outer_array_shape(observation,
51-
self._input_tensor_spec)
50+
shape = nest_utils.get_outer_array_shape(inputs, self._input_tensor_spec)
5251
return tf.constant(
5352
self._constant_output_val, tf.float32, shape=shape), network_state

0 commit comments

Comments
 (0)