| 1234567891011121314151617181920212223242526272829303132 |
- # Copyright (c) Alibaba, Inc. and its affiliates.
- from modelscope.hub.api import HubApi
- class DatasetDeleteManager(object):
- def __init__(self, dataset_name: str, namespace: str, version: str):
- self.api = HubApi()
- self.dataset_name = dataset_name
- self.namespace = namespace
- self.version = version
- def delete(self, object_name: str) -> str:
- # single object
- if not object_name.endswith('/'):
- resp_msg = self.api.delete_oss_dataset_object(
- object_name=object_name,
- dataset_name=self.dataset_name,
- namespace=self.namespace,
- revision=self.version)
- else:
- # multiple objects
- object_name = object_name.strip('/')
- resp_msg = self.api.delete_oss_dataset_dir(
- object_name=object_name,
- dataset_name=self.dataset_name,
- namespace=self.namespace,
- revision=self.version)
- return resp_msg
|
