From 897a86f4a57466d339ebc24a0256d3df5be09f29 Mon Sep 17 00:00:00 2001 From: Alvaro Sanchez-Gonzalez Date: Mon, 16 Nov 2020 12:10:09 +0000 Subject: [PATCH] Documentation fixes. PiperOrigin-RevId: 342606143 --- learning_to_simulate/learned_simulator.py | 2 +- learning_to_simulate/noise_utils.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/learning_to_simulate/learned_simulator.py b/learning_to_simulate/learned_simulator.py index 20f3728..2efeef8 100644 --- a/learning_to_simulate/learned_simulator.py +++ b/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] diff --git a/learning_to_simulate/noise_utils.py b/learning_to_simulate/noise_utils.py index 0c21599..ce06698 100644 --- a/learning_to_simulate/noise_utils.py +++ b/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)