TFTS: Make estimators_test non-flaky
authorAllen Lavoie <allenl@google.com>
Wed, 9 May 2018 18:28:30 +0000 (11:28 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Wed, 9 May 2018 20:08:46 +0000 (13:08 -0700)
commit80ec58f7d6f59618aaf7da7e0465441c7c83bc1d
treecd7135f89474d8913ddacefe060bf765351bbe5a
parent7baa9ffe735adfa11c987c435216943767530269
TFTS: Make estimators_test non-flaky

Replaces a "loss decreased" check with basic shape checking (it should have been seeded already, so there's likely some race condition which I should track down...).

PiperOrigin-RevId: 196001526
tensorflow/contrib/timeseries/python/timeseries/estimators_test.py