5f01692afeac101a3d5ff8364474a8e360f7cca1,tf_agents/utils/common_test.py,SoftVariablesUpdateTest,testShuffleOrderVariables,#SoftVariablesUpdateTest#Any#,94

Before Change

  @parameterized.parameters(0.0, 0.5, 1.0)
  def testShuffleOrderVariables(self, tau):
    inputs = tf.constant([[1, 2], [3, 4]], dtype=tf.float32)
    tf.contrib.layers.fully_connected(inputs, 2, scope="source")
    tf.contrib.layers.fully_connected(inputs, 2, scope="target")

    source_vars = tf.contrib.framework.get_model_variables("source")
    target_vars = tf.contrib.framework.get_model_variables("target")

After Change

  def testShuffleOrderVariables(self, tau):
    inputs = tf.constant([[1, 2], [3, 4]], dtype=tf.float32)
    source_net = tf.keras.layers.Dense(2, name="source_net")
    target_net = tf.keras.layers.Dense(2, name="target_net")

    // Force variable creation
    source_net(inputs)
    target_net(inputs)

    source_vars = source_net.trainable_weights
    target_vars = target_net.trainable_weights

    shuffled_source_vars = sorted(source_vars,
                                  key=lambda x: random.random())
    shuffled_target_vars = sorted(target_vars,

Instances