Documentation fixes.

PiperOrigin-RevId: 342606143

learning_to_simulate/learned_simulator.py

@@ -207,7 +207,7 @@ class LearnedSimulator(snt.AbstractModule):
   def get_predicted_and_target_normalized_accelerations(
       self, next_position, position_sequence_noise, position_sequence,
       n_particles_per_example, global_context=None, particle_types=None):  # pylint: disable=g-doc-args
-    """Produces a model step, outputting the next position for each particle.
+    """Produces normalized and predicted acceleration targets.
 
     Args:
       next_position: Tensor of shape [num_particles_in_batch, num_dimensions]

learning_to_simulate/noise_utils.py

@@ -45,8 +45,8 @@ def get_random_walk_noise_for_position_sequence(
 
   # Integrate the noise in the velocity to the positions, assuming
   # an Euler intergrator and a dt = 1, and adding no noise to the very first
-  # position (since that will only be used to calculate the first position#
-  # change.
+  # position (since that will only be used to calculate the first position
+  # change).
   position_sequence_noise = tf.concat([
       tf.zeros_like(velocity_sequence_noise[:, 0:1]),
       tf.cumsum(velocity_sequence_noise, axis=1)], axis=1)