def test_custom_estimator(self):
     custom_estimator.main([None, "--train_steps=1"])
Example #2
0
 def test_custom_estimator(self):
   custom_estimator.main([None, "--train_steps=1"])