Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion swanlab/api/openapi/group.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ def __init__(self, http: ApiHTTP):
super().__init__(http)

def list_workspaces(self) -> ApiResponse[list]:
resp = self.http.get("/group/")
resp = self.http.get("/group/", params={})
if resp.errmsg:
return resp
groups = resp.data.get("list", [])
Expand Down
24 changes: 6 additions & 18 deletions test/unit/api/openapi/test_experiment.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,19 +12,7 @@

import tutils as T
from swanlab import OpenApi
from swanlab.api.openapi.types import ApiResponse, Experiment, Pagination


@pytest.mark.skipif(T.is_skip_cloud_test, reason="skip cloud test")
def test_get_exp_state():
"""
获取一个实验的状态
"""
api = OpenApi()
res = api.get_exp_state(project="test_project", exp_cuid="test_cuid")
assert isinstance(res, ApiResponse)
if res.code == 200:
assert res.state == "FINISHED" or res.state == "RUNNING"
from swanlab.api.openapi.types import ApiResponse, Experiment


@pytest.mark.skipif(T.is_skip_cloud_test, reason="skip cloud test")
Expand All @@ -34,32 +22,32 @@ def test_get_experiment():
"""
api = OpenApi()
exp_cuid = "test_cuid"
res = api.get_experiment(project="test_project", exp_cuid=exp_cuid)
res = api.get_experiment(project="test_project", exp_id=exp_cuid)
assert isinstance(res, ApiResponse)
if res.code == 200:
assert isinstance(res.data, Experiment)

@pytest.mark.skipif(T.is_skip_cloud_test, reason="skip cloud test")
def test_list_project_exps():
def test_list_experiments():
"""
获取一个项目下的实验列表
"""
api = OpenApi()
res = api.list_project_exps(project="SwanLab")
res = api.list_experiments(project="SwanLab")
assert isinstance(res, ApiResponse)
if res.code == 200:
assert isinstance(res.data, list)
for item in res.data:
assert isinstance(item, Experiment)

@pytest.mark.skipif(T.is_skip_cloud_test, reason="skip cloud test")
def test_get_exp_summary():
def test_get_summary():
"""
获取一个实验的Summary信息
"""
api = OpenApi()
exp_cuid = "test_cuid"
res = api.get_exp_summary(project="test_project", exp_cuid=exp_cuid)
res = api.get_summary(project="test_project", exp_id=exp_cuid)
assert isinstance(res, ApiResponse)
if res.code == 200:
assert isinstance(res.data, dict)
2 changes: 0 additions & 2 deletions test/unit/api/openapi/test_types.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,6 @@
测试开放API模型定义
"""

import pytest

import tutils as T
from swanlab.api.openapi.types import Experiment

Expand Down