Searched refs:mock_est (Results 1 – 1 of 1) sorted by relevance
318 mock_est = test.mock.Mock(spec=estimator_lib.Estimator)325 training.train_and_evaluate(mock_est, mock_train_spec, mock_eval_spec)326 mock_executor.assert_called_with(estimator=mock_est,342 mock_est = test.mock.Mock(spec=estimator_lib.Estimator)343 mock_est.config = _create_run_config_with_cluster_spec(tf_config)348 training.train_and_evaluate(mock_est, mock_train_spec, mock_eval_spec)433 mock_est = test.mock.Mock(spec=estimator_lib.Estimator)434 mock_est.config = self._run_config440 executor = training._TrainingExecutor(mock_est, train_spec, mock_eval_spec)444 mock_est.config.cluster_spec,[all …]