def branch(self, branch):
self._branch = branch
if self._branch not in self.list_branches():
self.create_branch(self._branch)
self.branch_path = get_experiment_path(self.path, self._branch)
self.index_path = get_experiment_run_path(self.path, self._branch,
self.active_commit)
self.objects_dir_path = get_run_objects_dir_path(self.path,
self._branch,
self.active_commit)
self.media_dir_path = os.path.join(self.objects_dir_path,
AIM_MEDIA_DIR_NAME)
self.meta_file_content = None
self.meta_file_path = get_run_objects_meta_file_path(self.path,
self._branch,
self.active_commit)
if not os.path.isdir(self.index_path):
os.makedirs(self.index_path)
if self.records_storage:
self.records_storage.close()
if os.path.exists(self.branch_path):
self.records_storage = self.get_records_storage(
self.objects_dir_path,
self.mode)
def open_storage(self):
if self._storage is None:
storage_path = get_run_objects_dir_path(self.repo.path,
self.experiment_name,
self.run_hash)
self._storage = self.repo.get_records_storage(storage_path, 'r')