Small change to trigger pipelines

pymc-devs · Jul 16, 2020 · ae4edf1 · ae4edf1
1 parent 9316657
commit ae4edf1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymc4/variational/util.py b/pymc4/variational/util.py
@@ -41,7 +41,7 @@ def split(self, flatten_tensor: tf.Tensor) -> Dict[str, tf.Tensor]:
             split_view[param] = tf.cast(tf.reshape(flatten_tensor[slc], shape), dtype)
         return split_view
 
-    def split_samples(self, samples, n):
+    def split_samples(self, samples: tf.Tensor, n: int):
         """Split view of samples after drawing samples from posterior."""
         q_samples = dict()
         for param in self.free_rvs.keys():