message: TFTS: axis=<list> -> axis=<tuple> in numpy.squeeze
  Apparently lists and tuples behave differently in some numpy versions.
diff: @@ -80,8 +80,8 @@ def multivariate_train_and_sample(
  session=session, steps=1))
  next_sample = numpy.random.multivariate_normal(
  # Squeeze out the batch and series length dimensions (both 1).
  - mean=numpy.squeeze(current_prediction["mean"], axis=[0, 1]),
  - cov=numpy.squeeze(current_prediction["covariance"], axis=[0, 1]))
  + mean=numpy.squeeze(current_prediction["mean"], axis=(0, 1)),
  + cov=numpy.squeeze(current_prediction["covariance"], axis=(0, 1)))
  # Update model state so that future predictions are conditional on the
  # value we just sampled.
  filtering_features = {
  