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
3 changes: 3 additions & 0 deletions swanlab/api/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,5 +10,8 @@
from .info import *
from .auth.login import terminal_login, code_login
from .http import create_http, get_http
import nest_asyncio

nest_asyncio.apply()

__all__ = ["LoginInfo", 'ExperimentInfo', 'ProjectInfo', "code_login", 'terminal_login', 'create_http', 'get_http']
5 changes: 2 additions & 3 deletions swanlab/data/sdk.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
import sys
import traceback
from datetime import datetime
from typing import Optional, Union, List, Dict
from typing import Optional, Union, Dict
from .modules import DataType
from .run import SwanLabRun, SwanLabConfig, register
from .utils.file import check_dir_and_create, formate_abs_path
Expand All @@ -27,7 +27,7 @@
from swanlab.api.upload.model import ColumnModel
from swanlab.package import version_limit, get_package_version, get_host_api, get_host_web
from swanlab.error import KeyFileError
from swanlab.cloud import LogSnifferTask, ThreadPool, UploadType
from swanlab.cloud import LogSnifferTask, ThreadPool
from swanlab.utils import create_time
from swanlab.cloud import UploadType

Expand Down Expand Up @@ -217,7 +217,6 @@ def _metric_callback(is_scalar: dict, data):
上传指标回调函数
:param is_scalar: 是否为标量指标
:param data: 上传的指标
:return:
"""
run.pool.queue.put((UploadType.SCALAR_METRIC if is_scalar else UploadType.MEDIA_METRIC, [data]))

Expand Down