Another fix for tf_doctest
@@ -1102,8 +1102,7 @@ class OptimizerV2(tf.__internal__.tracking.Trackable):
>>> m.compile(opt, loss='mse')
>>> data = np.arange(100).reshape(5, 20)
>>> labels = np.zeros(5)
- >>> print('Training'); results = m.fit(data, labels)
- Training ...
+ >>> results = m.fit(data, labels) # Training.
>>> len(opt.get_weights())
3
@@ -1133,8 +1132,7 @@ class OptimizerV2(tf.__internal__.tracking.Trackable):
>>> m.compile(opt, loss='mse')
>>> data = np.arange(100).reshape(5, 20)
>>> labels = np.zeros(5)
- >>> print('Training'); results = m.fit(data, labels)
- Training ...
+ >>> results = m.fit(data, labels) # Training.
>>> new_weights = [np.array(10), np.ones([20, 10]), np.zeros([10])]
>>> opt.set_weights(new_weights)
>>> opt.iterations
