From 2d3d6cb01820ff942a89441389384f63fdf22593 Mon Sep 17 00:00:00 2001 From: Alvaro Sanchez-Gonzalez Date: Wed, 11 Nov 2020 12:01:52 +0000 Subject: [PATCH] Documentation fix. PiperOrigin-RevId: 341801256 --- learning_to_simulate/learned_simulator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/learning_to_simulate/learned_simulator.py b/learning_to_simulate/learned_simulator.py index a8f19f2..20f3728 100644 --- a/learning_to_simulate/learned_simulator.py +++ b/learning_to_simulate/learned_simulator.py @@ -98,8 +98,8 @@ class LearnedSimulator(snt.AbstractModule): If None, we assume all particles are the same type. Returns: - updated_position_sequence with shape [num_particles_in_batch, - num_dimensions] for one step into the future from the input sequence. + Next position with shape [num_particles_in_batch, num_dimensions] for one + step into the future from the input sequence. """ input_graphs_tuple = self._encoder_preprocessor( position_sequence, n_particles_per_example, global_context,