def test_get_tag(self):
mdpath = os.path.join(self.test_dir, 'metadata')
specpath = 'dataset-ex'
ensure_path_exists(os.path.join(mdpath, specpath))
shutil.copy('hdata/dataset-ex.spec', os.path.join(mdpath, specpath) + '/dataset-ex.spec')
manifestpath = os.path.join(os.path.join(mdpath, specpath), 'MANIFEST.yaml')
yaml_save(files_mock, manifestpath)
config['mlgit_path'] = self.test_dir
m = Metadata(specpath, mdpath, config, DATASETS)
r = Repository(config, DATASETS)
r.init()
tag_list = ['computer__images__dataset-ex__1']
with mock.patch('ml_git.metadata.Metadata.list_tags', return_value=tag_list):
target_tag = m.get_tag(specpath, -1)
self.assertEqual(target_tag, tag_list[0])
clear(m.path)
def test_tag_exist(self):
mdpath = os.path.join(self.test_dir, 'metadata')
specpath = 'dataset-ex'
ensure_path_exists(os.path.join(mdpath, specpath))
shutil.copy('hdata/dataset-ex.spec',
os.path.join(mdpath, specpath) + '/dataset-ex.spec')
manifestpath = os.path.join(os.path.join(mdpath, specpath),
'MANIFEST.yaml')
yaml_save(files_mock, manifestpath)
config['mlgit_path'] = self.test_dir
m = Metadata(specpath, mdpath, config, repotype)
r = Repository(config, repotype)
r.init()
fullmetadatapath, categories_subpath, metadata = m.tag_exists(
self.test_dir)
self.assertFalse(metadata is None)
def init(entity):
"""This command will start the ml-git entity.
Examples:
init('repository')
init('dataset')
Args:
entity (str): The type of entity that will be initialized (repository, dataset, labels or model).
"""
if entity == 'repository':
init_mlgit()
elif entity in EntityType.to_list():
repo = Repository(config_load(), entity)
repo.init()
else:
log.error('The type of entity entered is invalid. Valid types are: [repository, dataset, labels or model]')