def test_metadata(self):
pb = self.text('do', 'A deer. A female deer.')
summary_metadata = pb.value[0].metadata
plugin_data = summary_metadata.plugin_data
self.assertEqual(summary_metadata.summary_description, '')
self.assertEqual(plugin_data.plugin_name, metadata.PLUGIN_NAME)
content = summary_metadata.plugin_data.content
# There's no content, so successfully parsing is fine.
metadata.parse_plugin_metadata(content)
def test_explicit_description(self):
description = 'A whole step above do.'
pb = self.text('re', 'A drop of golden sun.', description=description)
summary_metadata = pb.value[0].metadata
self.assertEqual(summary_metadata.summary_description, description)
plugin_data = summary_metadata.plugin_data
self.assertEqual(plugin_data.plugin_name, metadata.PLUGIN_NAME)
content = summary_metadata.plugin_data.content
# There's no content, so successfully parsing is fine.
metadata.parse_plugin_metadata(content)
def test_metadata(self):
pb = self.compute_and_check_summary_pb('do', 'A deer. A female deer.')
summary_metadata = pb.value[0].metadata
plugin_data = summary_metadata.plugin_data
self.assertEqual(summary_metadata.display_name, 'do')
self.assertEqual(summary_metadata.summary_description, '')
self.assertEqual(plugin_data.plugin_name, metadata.PLUGIN_NAME)
content = summary_metadata.plugin_data.content
# There's no content, so successfully parsing is fine.
metadata.parse_plugin_metadata(content)
def test_explicit_display_name_and_description(self):
display_name = '"Re"'
description = 'A whole step above do.'
pb = self.compute_and_check_summary_pb('re', 'A drop of golden sun.',
display_name=display_name,
description=description)
summary_metadata = pb.value[0].metadata
self.assertEqual(summary_metadata.display_name, display_name)
self.assertEqual(summary_metadata.summary_description, description)
plugin_data = summary_metadata.plugin_data
self.assertEqual(plugin_data.plugin_name, metadata.PLUGIN_NAME)
content = summary_metadata.plugin_data.content
# There's no content, so successfully parsing is fine.
metadata.parse_plugin_metadata(content)
def index_impl(self, ctx, experiment):
mapping = self._data_provider.list_tensors(
ctx,
experiment_id=experiment,
plugin_name=metadata.PLUGIN_NAME,
)
result = {run: [] for run in mapping}
for (run, tag_to_content) in mapping.items():
for (tag, metadatum) in tag_to_content.items():
md = metadata.parse_plugin_metadata(metadatum.plugin_content)
if not self._version_checker.ok(md.version, run, tag):
continue
result[run].append(tag)
return result
