def test_display_tensorboard(self, mock_open, mock_os, mock_json): mock_os.path.isfile.return_value = False display.display(display.Tensorboard('gs://job/dir')) mock_json.dump.assert_called_with( {'outputs': [{ 'type': 'tensorboard', 'source': 'gs://job/dir' }]}, mock.ANY)
def _dump_job_metadata(project_id, job_id, job): display.display( display.Link( 'https://console.cloud.google.com/mlengine/jobs/{}?project={}'. format(job_id, project_id), 'Job Details')) display.display( display.Link( 'https://console.cloud.google.com/logs/viewer?project={}&resource=ml_job/job_id/{}&interval=NO_LIMIT' .format(project_id, job_id), 'Logs')) if 'trainingInput' in job and 'jobDir' in job['trainingInput']: display.display(display.Tensorboard(job['trainingInput']['jobDir']))
def test___repr__(self, mock_open, mock_os, mock_json): self.assertEqual('# Title', str(display.Markdown('# Title'))) self.assertEqual('Open Tensorboard at: gs://trained/model/', str(display.Tensorboard('gs://trained/model/'))) self.assertEqual('title: https://test/uri', str(display.Link('https://test/uri', 'title')))