From 1ea71cb1bf969bbf4cdae3ce26f5c5e2702f64eb Mon Sep 17 00:00:00 2001
From: Rishi Verma <rishioffic@gmail.com>
Date: Wed, 24 Sep 2025 05:20:34 +0000
Subject: [PATCH 1/3] build: added grouping to config

---
 src/state/configs/wandb/default.yaml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/state/configs/wandb/default.yaml b/src/state/configs/wandb/default.yaml
index 0d13940a..a773a9a9 100644
--- a/src/state/configs/wandb/default.yaml
+++ b/src/state/configs/wandb/default.yaml
@@ -3,4 +3,5 @@
 entity: your_entity_name
 project: state
 local_wandb_dir: ./wandb_logs
-tags: [] 
+tags: []
+group: null

From d29b245b436df69f62497cee791efe916f1de6af Mon Sep 17 00:00:00 2001
From: Rishi Verma <rishioffic@gmail.com>
Date: Wed, 24 Sep 2025 05:20:57 +0000
Subject: [PATCH 2/3] build: added optional group to wandb logger

---
 src/state/tx/utils/__init__.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/state/tx/utils/__init__.py b/src/state/tx/utils/__init__.py
index 20d12c78..3541308a 100644
--- a/src/state/tx/utils/__init__.py
+++ b/src/state/tx/utils/__init__.py
@@ -1,10 +1,10 @@
 import time
 import logging
+import os
 from contextlib import contextmanager
 from lightning.pytorch.loggers import CSVLogger, WandbLogger
 from lightning.pytorch.loggers.csv_logs import CSVLogger as BaseCSVLogger
 import csv
-import os
 from lightning.pytorch.callbacks import ModelCheckpoint
 from os.path import join
 
@@ -100,13 +100,14 @@ def get_loggers(
         try:
             # Check if wandb is available
             import wandb
-
+            
             wandb_logger = WandbLogger(
                 name=name,
                 project=wandb_project,
                 entity=wandb_entity,
                 dir=local_wandb_dir,
                 tags=cfg["wandb"].get("tags", []) if cfg else [],
+                group=cfg["wandb"].get("group", None) if cfg else None,
             )
             if cfg is not None:
                 wandb_logger.experiment.config.update(cfg)

From 8fe1e3e310e8ae1dfc270dcf4686d517d2ac8aff Mon Sep 17 00:00:00 2001
From: Rishi Verma <rishioffic@gmail.com>
Date: Wed, 24 Sep 2025 05:42:52 +0000
Subject: [PATCH 3/3] chore: updated version

---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index dec4906e..e6eb7541 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "arc-state"
-version = "0.9.31"
+version = "0.9.32"
 description = "State is a machine learning model that predicts cellular perturbation response across diverse contexts."
 readme = "README.md"
 authors = [