TFTS: axis=<list> -> axis=<tuple> in numpy.squeeze

Apparently lists and tuples behave differently in some numpy versions. PiperOrigin-RevId: 206616759
willsim · Jul 30, 2018 · 7bab62b · 7bab62b
1 parent b3df17e
commit 7bab62b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensorflow/contrib/timeseries/examples/multivariate.py b/tensorflow/contrib/timeseries/examples/multivariate.py
@@ -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 = {