Fixes for downstream integration: (#458)

- np.Inf becomes np.inf
- explicit `module` for a test gin-configurable

Author: mtrofin

compiler_opt/rl/imitation_learning/generate_bc_trajectories_lib.py

@@ -542,8 +542,8 @@ def explore_at_state_generator(
 
     distr_logits = explore_policy(explore_state).action.logits.numpy()[0]
     for _ in range(num_samples):
-      distr_logits[replay_prefix[explore_step]] = -np.Inf
-      if all(-np.Inf == logit for logit in distr_logits):
+      distr_logits[replay_prefix[explore_step]] = -np.inf
+      if all(-np.inf == logit for logit in distr_logits):
         break
       replay_prefix[explore_step] = np.random.choice(
           range(distr_logits.shape[0]), p=scipy.special.softmax(distr_logits))

compiler_opt/rl/imitation_learning/generate_bc_trajectories_test.py

@@ -453,9 +453,9 @@ def _explore_policy(state: time_step.TimeStep):
       # will explore every 4-th step
       logits = [[
           4.0 + 1e-3 * float(env_test._NUM_STEPS - times_called),
-          -np.Inf,
-          -np.Inf,
-          -np.Inf,
+          -np.inf,
+          -np.inf,
+          -np.inf,
           float(np.mod(times_called, 5)),
       ]]
       return policy_step.PolicyStep(
@@ -632,7 +632,7 @@ def test_process_succeeded(self):
     self.assertEqual(seq_example, succeeded_comp[0][0][2])
 
 
-@gin.configurable
+@gin.configurable(module='generate_bc_trajectories_test')
 class MockModuleWorker(generate_bc_trajectories_lib.ModuleWorker):
 
   @mock.patch('subprocess.Popen')