Searched refs:mock_latest_checkpoint (Results 1 – 1 of 1) sorted by relevance
321 def test_validation_monitor_no_ckpt(self, mock_latest_checkpoint, argument326 mock_latest_checkpoint.return_value = None336 mock_latest_checkpoint.assert_called_with(model_dir)341 mock_latest_checkpoint, argument347 mock_latest_checkpoint.return_value = '%s/ckpt' % model_dir360 def test_validation_monitor_invalid_metric(self, mock_latest_checkpoint, argument366 mock_latest_checkpoint.return_value = '%s/ckpt' % model_dir379 def test_validation_monitor(self, mock_latest_checkpoint, argument402 mock_latest_checkpoint.return_value = '%s/ckpt.%s' % (model_dir, step)425 mock_latest_checkpoint.return_value = '%s/ckpt.%s' % (model_dir, step)[all …]