internal merge of PR #1190

PiperOrigin-RevId: 219710898

Author: koz4k

tensor2tensor/layers/modalities.py

@@ -143,7 +143,7 @@ def top(self, body_output, _):
 
     if self._model_hparams.shared_embedding_and_softmax_weights:
       scope_name = "shared"
-      reuse = tf.AUTO_REUSE
+      reuse = True
     else:
       scope_name = "softmax"
       reuse = False

tensor2tensor/rl/trainer_model_based.py

@@ -83,6 +83,7 @@ def world_model_step_increment(hparams, is_initial_epoch):
 
 
 def setup_directories(base_dir, subdirs):
+  """Setup directories."""
   base_dir = os.path.expanduser(base_dir)
   tf.gfile.MakeDirs(base_dir)
 
@@ -423,7 +424,7 @@ def initial_frame_chooser(batch_size):
     def decode_real_obs(index):
       return np.stack([
           subsequence[index].observation.decode()
-          for subsequence in eval_subsequences
+          for subsequence in eval_subsequences  # pylint: disable=cell-var-from-loop
       ])
     real_init_obs = decode_real_obs(0)
     assert np.all(sim_init_obs == real_init_obs)