44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
# These codes are copied from modelscope revision c58451baead80d83281f063d12fb377fad415257
|
|
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
|
|
from http.cookiejar import CookieJar
|
|
from typing import Tuple
|
|
|
|
|
|
class BaseAuthConfig(object):
|
|
"""Base authorization config class."""
|
|
|
|
def __init__(self, cookies: CookieJar, git_token: str,
|
|
user_info: Tuple[str, str]):
|
|
self.cookies = cookies
|
|
self.git_token = git_token
|
|
self.user_info = user_info
|
|
|
|
|
|
class OssAuthConfig(BaseAuthConfig):
|
|
"""The authorization config for oss dataset."""
|
|
|
|
def __init__(self, cookies: CookieJar, git_token: str,
|
|
user_info: Tuple[str, str]):
|
|
super().__init__(
|
|
cookies=cookies, git_token=git_token, user_info=user_info)
|
|
|
|
|
|
class VirgoAuthConfig(BaseAuthConfig):
|
|
"""The authorization config for virgo dataset."""
|
|
|
|
def __init__(self, cookies: CookieJar, git_token: str,
|
|
user_info: Tuple[str, str]):
|
|
super().__init__(
|
|
cookies=cookies, git_token=git_token, user_info=user_info)
|
|
|
|
|
|
class MaxComputeAuthConfig(BaseAuthConfig):
|
|
# TODO: MaxCompute dataset to be supported.
|
|
def __init__(self, cookies: CookieJar, git_token: str,
|
|
user_info: Tuple[str, str]):
|
|
super().__init__(
|
|
cookies=cookies, git_token=git_token, user_info=user_info)
|
|
|
|
self.max_compute_grant_cmd = None
|