code: def testLegalActionsMask(self):
      action_values = [0.9, 1.0, 0.9, 0.1, -0.6]
      legal_actions_mask = [0.0, 1.0, 1.0, 1.0, 1.0]
      epsilon = 0.1
  
      expected = [0.00, 0.925, 0.025, 0.025, 0.025]
  
      result = policy_ops.epsilon_greedy(
          action_values, epsilon, legal_actions_mask
      ).probs
      with self.test_session() as sess:
          self.assertAllClose(sess.run(result), expected)
  
file_path: trfl/trfl/policy_ops_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow.compat.v1
  from trfl import policy_ops
label_desc: Use image node_id 6 for calling the EpsilonGreedyTest obj's underlying member method code with example usage: obj.testLegalActionsMask() without return types