|
|
from argparse import ArgumentParser |
|
|
from typing import Optional |
|
|
|
|
|
from datasets.commands import BaseDatasetsCLICommand |
|
|
from datasets.hub import delete_from_hub |
|
|
|
|
|
|
|
|
def _command_factory(args): |
|
|
return DeleteFromHubCommand( |
|
|
args.dataset_id, |
|
|
args.config_name, |
|
|
args.token, |
|
|
args.revision, |
|
|
) |
|
|
|
|
|
|
|
|
class DeleteFromHubCommand(BaseDatasetsCLICommand): |
|
|
@staticmethod |
|
|
def register_subcommand(parser): |
|
|
parser: ArgumentParser = parser.add_parser("delete_from_hub", help="Delete dataset config from the Hub") |
|
|
parser.add_argument( |
|
|
"dataset_id", help="source dataset ID, e.g. USERNAME/DATASET_NAME or ORGANIZATION/DATASET_NAME" |
|
|
) |
|
|
parser.add_argument("config_name", help="config name to delete") |
|
|
parser.add_argument("--token", help="access token to the Hugging Face Hub") |
|
|
parser.add_argument("--revision", help="source revision") |
|
|
parser.set_defaults(func=_command_factory) |
|
|
|
|
|
def __init__( |
|
|
self, |
|
|
dataset_id: str, |
|
|
config_name: str, |
|
|
token: Optional[str], |
|
|
revision: Optional[str], |
|
|
): |
|
|
self._dataset_id = dataset_id |
|
|
self._config_name = config_name |
|
|
self._token = token |
|
|
self._revision = revision |
|
|
|
|
|
def run(self) -> None: |
|
|
_ = delete_from_hub(self._dataset_id, self._config_name, revision=self._revision, token=self._token) |
|
|
|
