update doc

Author: SuperXiang

pahelix/model_zoo/seq_vae_model.py

@@ -151,8 +151,8 @@ def sample_z_prior(self, n_batch):
         Args:
             n_batch: number of batches
 
-        Returns: 
-        (n_batch, d_z) of floats, sample of latent z
+        Returns:
+            (n_batch, d_z) of floats, sample of latent z
         """
         return paddle.randn([n_batch, self.q_mu.weight.shape[1]])
 
@@ -166,16 +166,15 @@ def tensor2string(self, tensor):
         return string
 
     def sample(self, n_batch, max_len=100, z=None, temp=1.0):
-        """Generating n_batch samples in eval mode (`z` could be
-            not on same device)
+        """Generating n_batch samples in eval mode (z could be not on same device)
 
         Args:
             n_batch: number of sentences to generate
             max_len: max len of samples
             z: (n_batch, d_z) of floats, latent vector z or None
             temp: temperature of softmax
 
-        Returns: 
+        Returns:
             list of tensors of strings, samples sequence x
         """