Skip to content

Conversation

SAKURA-CAT
Copy link
Member

@SAKURA-CAT SAKURA-CAT commented Mar 2, 2025

此次拉取请求(Pull Request)为 swanlab 项目引入了多项更改,主要集中在添加对捕获和处理 Conda 环境信息的支持。这些更改涉及多个文件,包括模型更新、元数据处理以及单元测试。

主要更改包括:

模型和元数据更新:

  • swanlab/api/upload/model.py 中的 FileModel 类中添加了一个新的 conda 属性,并更新了 create 方法以处理这个新属性。[1] [2]
  • 更新了 swanlab/data/callback_cloud.py 中的 on_runtime_info_update 方法,使其在创建 FileModel 实例时包含 Conda 信息。
  • 修改了 swanlab/data/run/main.py,以导入并使用新的 get_conda 函数来获取 Conda 信息。[1] [2]
  • 更新了 swanlab/data/run/metadata/__init__.py,以包含新的 get_conda 函数,并相应地调整了 __all__ 列表。[1] [2]

新增功能:

  • 新增了 swanlab/data/run/metadata/conda.py 文件,其中定义了 get_conda 函数,该函数通过子进程和 YAML 解析来收集 Conda 环境信息。

单元测试:

  • 更新了 test/unit/api/upload/test_model.py 中的单元测试,以测试 FileModel 类中的新 conda 属性。[1] [2] [3]
  • 新增了 test/unit/data/run/metadata/hardware/test_conda.py 测试文件,用于测试 get_conda 函数,确保其正确获取并解析 Conda 环境信息。

close #827

chore: opt some import
@SAKURA-CAT SAKURA-CAT self-assigned this Mar 2, 2025
@SAKURA-CAT SAKURA-CAT added 🐛 bug Something isn't working 💪 enhancement New feature or request labels Mar 2, 2025
@SwanHubX SwanHubX deleted a comment from SAKURA-CAT Mar 3, 2025
@Zeyi-Lin
Copy link
Member

Zeyi-Lin commented Mar 3, 2025

鉴于conda env export在实际环境下极长的导出时间,需要在swanlab.init中传入例如save_conda,或swanlab.Settings.save_conda这样的参数时,才会启动保存逻辑。

@Zeyi-Lin Zeyi-Lin self-requested a review March 4, 2025 05:18
@SAKURA-CAT SAKURA-CAT marked this pull request as ready for review March 4, 2025 05:21
@SAKURA-CAT SAKURA-CAT merged commit 87b1db8 into main Mar 4, 2025
5 checks passed
@SAKURA-CAT SAKURA-CAT deleted the feat/conda branch March 4, 2025 05:21
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

🐛 bug Something isn't working 💪 enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[REQUEST] 环境-记录conda环境

2 participants