import json
from opendatalab.__version__ import __url__
from opendatalab.cli.get import implement_get
from opendatalab.cli.info import implement_info
from opendatalab.cli.login import implement_login
from opendatalab.cli.ls import implement_ls
from opendatalab.cli.search import implement_search
from opendatalab.cli.utility import ContextInfo
if __name__ == '__main__':
"""
ContextInfo: default
please use shell login first, use: opendatalab login
"""
ctx = ContextInfo(__url__, "")
client = ctx.get_client()
odl_api = client.get_api()
# 0. login with account
# account = "xxxxx" # your username
# pw = "xxxxx" # your password
# print(f'*****'*8)
# implement_login(ctx, account, pw)
# 1. search demo
res_list = odl_api.search_dataset("OpenXD-OmniObject3D-New")
for index, res in enumerate(res_list):
print(f"index: {index}, result: {res['name']}")
# implement_search("coco")
print(f'*****'*8)
# 2. list demo
implement_ls(ctx, 'OpenXD-OmniObject3D-New')
print(f'*****' * 8)
# 4. get dataset info
implement_info(ctx, 'OpenXD-OmniObject3D-New')
# 5. download
# get all files of dataset
# implement_get(ctx, "MNIST", 4, 0)
# get partial files of dataset
implement_get(ctx, "OpenXD-OmniObject3D-New/raw_scans", 4, 0)
print(f'*****' * 5)
****************************************
Traceback (most recent call last):
File "/share/liangyingping/dataset/omni/omni.py", line 37, in <module>
implement_info(ctx, 'OpenXD-OmniObject3D-New')
File "/data/liangyingping/anaconda3/envs/sketch_diffusion/lib/python3.9/site-packages/opendatalab/cli/utility.py", line 139, in wrapper
func(*args, **kwargs)
File "/data/liangyingping/anaconda3/envs/sketch_diffusion/lib/python3.9/site-packages/opendatalab/cli/info.py", line 82, in implement_info
info_data_result = reformat_info_data(info_data)
File "/data/liangyingping/anaconda3/envs/sketch_diffusion/lib/python3.9/site-packages/opendatalab/cli/info.py", line 38, in reformat_info_data
citation_data = info_data['attrs']['citation']
KeyError: 'citatio