mirror of
https://gitcode.com/gh_mirrors/eas/EasyFace.git
synced 2026-01-28 03:00:16 +00:00
32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
# 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
|