From 265fb609676ea6ba8d7eacd0623bf68165790615 Mon Sep 17 00:00:00 2001
From: Janosh Riebesell <janosh.riebesell@gmail.com>
Date: Wed, 12 Jun 2024 18:50:29 -0400
Subject: [PATCH] add optional deps set logging = ["wandb>=0.17"] and install
 in CI

---
 .github/workflows/test.yml | 2 +-
 .gitignore                 | 3 +++
 pyproject.toml             | 1 +
 3 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index c4fabe05..b6a12480 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -38,7 +38,7 @@ jobs:
 
           python setup.py build_ext --inplace
 
-          uv pip install -e .[test] --system --resolution=${{ matrix.version.resolution }}
+          uv pip install -e .[test,logging] --system --resolution=${{ matrix.version.resolution }}
 
       - name: Run Tests
         run: pytest --capture=no --cov --cov-report=xml
diff --git a/.gitignore b/.gitignore
index aea32c2b..ac65ad05 100644
--- a/.gitignore
+++ b/.gitignore
@@ -27,3 +27,6 @@ coverage.xml
 .ipynb_checkpoints
 bond_graph_error.cif
 test.py
+
+# training logs
+wandb
diff --git a/pyproject.toml b/pyproject.toml
index dc6c0f49..243ecd87 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -32,6 +32,7 @@ test = ["pytest-cov>=4", "pytest>=8"]
 # needed to run interactive example notebooks
 examples = ["crystal-toolkit>=2023.11.3", "pandas>=2.2"]
 docs = ["lazydocs>=0.4"]
+logging = ["wandb>=0.17"]
 
 [project.urls]
 Source = "https://github.com/CederGroupHub/chgnet"