def get_last_entity_version(entity_type, entity_name):
config = merged_config_load()
metadata_path = get_metadata_path(config, entity_type)
metadata = Metadata(entity_name, metadata_path, config, entity_type)
if not metadata.check_exists():
log.error(output_messages['ERROR_NOT_INITIALIZED'] % entity_type)
return
last_version = metadata.get_last_tag_version(entity_name)
return last_version + 1
def test_last_tag_version(self):
sepc_path = 'dataset-ex'
config['mlgit_path'] = self.test_dir
m = Metadata('', '', config, DATASETS)
m.init()
tag_list = [
'computer__images__dataset-ex__1',
'computer__images__dataset-ex__2'
]
with mock.patch('ml_git.metadata.Metadata.list_tags',
return_value=tag_list):
last_version = m.get_last_tag_version(sepc_path)
self.assertEqual(last_version, 2)
tag_list = []
with mock.patch('ml_git.metadata.Metadata.list_tags',
return_value=tag_list):
last_version = m.get_last_tag_version(sepc_path)
self.assertEqual(last_version, 0)
clear(self.test_dir)
