Claude Pytorch Treehugger

🚀 PyTorch HUD API （MCP サポート付き）Python ライブラリ

この Python ライブラリは PyTorch HUD API と MCP サーバーとのやり取りを可能にし、CI/CD データ、ジョブログ、および分析をサポートします。

🚀 クイックスタート

このプロジェクトは、PyTorch の継続的インテグレーション/デリバリー（CI/CD）データ分析に以下のツールを提供します。

• ワークフロー、ジョブ、およびテスト実行のデータを取得する
• 大規模な CI ログを効率的に分析する
• 分析用の ClickHouse クエリを統合する
• リソース利用率指標を取得する

📦 インストール

ユーザー向け

# GitHub リポジトリからインストール
pip install git+https://github.com/izaitsevfb/claude-pytorch-treehugger.git

claude mcp add hud pytorch-hud

開発者向け

# 依存関係をインストールする（pip でインストールされていない場合）
pip install -r requirements.txt

# MCP サーバーを起動する
python -m pytorch_hud

✨ 主な機能

データアクセス

• get_commit_summary：ジョブを含まない基本的なコミット情報を取得します。
• get_job_summary：集計されたジョブ状態統計を取得します。
• get_filtered_jobs：状態、ワークフロー、または名前に基づいてジョブをフィルタリングします。
• get_failure_details：失敗したジョブの詳細な失敗情報を取得します。
• get_recent_commit_status：最近のコミットの状態とジョブ統計データを取得します。

ログ分析

• download_log_to_file：ログをローカルストレージにダウンロードします。
• extract_log_patterns：エラー、警告などの情報を検索します。
• extract_test_results：テスト実行結果を解析します。
• filter_log_sections：特定のログセクションを抽出します。
• search_logs：複数のログで検索を行います。

💻 使用例

開発

# テストを実行する
python -m unittest discover test

# 型チェックを行う
mypy -p pytorch_hud -p test

# コードレビューを行う
ruff check pytorch_hud/ test/

📚 ドキュメント

• CLAUDE.md：詳細な使用方法、コードスタイル、および実装に関する注意事項が記載されています。
• mcp-guide.md：MCP プロトコルの一般的な情報が記載されています。

📄 ライセンス

MIT