def branches(self) -> list:
"""
Gets a list all the branches of the dataset
"""
if self._commit_id is None:
raise VersioningNotSupportedException("branches")
return self._branch_node_map.keys()
def log(self):
"""| Prints the commits in the commit tree before the current commit
Only works if dataset was created on or after Hub v1.3.0
"""
if self._commit_id is None:
raise VersioningNotSupportedException("log")
current_node = (self._version_node.parent
if not self._version_node.children else
self._version_node)
print(f"\n Current Branch: {self._branch}\n")
while current_node:
print(f"{current_node}\n")
current_node = current_node.parent
def checkout(self, address: str, create: bool = False) -> str:
"""| Changes the state of the dataset to the address mentioned. Creates a new branch if address isn't a commit id or branch name and create is True.
Always checks out to the head of a branch if the address specified is a branch name.
Returns the commit id of the commit that has been switched to.
Only works if dataset was created on or after Hub v1.3.0
Parameters
----------
address: str
The branch name or commit id to checkout to
create: bool, optional
Specifying create as True creates a new branch from the current commit if the address isn't an existing branch name or commit id
"""
if self._commit_id is None:
raise VersioningNotSupportedException("checkout")
self.flush()
if address in self._branch_node_map.keys():
self._branch = address
self._version_node = self._branch_node_map[address]
self._commit_id = self._version_node.commit_id
elif address in self._commit_node_map.keys():
self._version_node = self._commit_node_map[address]
self._branch = self._version_node.branch
self._commit_id = self._version_node.commit_id
elif create:
if "r" in self._mode:
raise ReadModeException("checkout to create new branch")
self._branch = address
new_commit_id = generate_hash()
new_node = VersionNode(new_commit_id, self._branch)
if not self._version_node.children:
for key in self.keys:
self._tensors[key].fs_map.copy_all_chunks(
self._commit_id, new_commit_id
)
if self._version_node.parent is not None:
self._version_node.parent.insert(
new_node, f"switched to new branch {address}"
)
else:
self._version_node.insert(new_node, f"switched to new branch {address}")
self._version_node = new_node
self._commit_id = new_commit_id
self._branch_node_map[self._branch] = new_node
self._commit_node_map[self._commit_id] = new_node
self.flush()
else:
raise AddressNotFound(address)
return self._commit_id