From 1e7728fd4eaddc438eec2862a11d4718455e6c24 Mon Sep 17 00:00:00 2001
From: harshitgavita-07 <harshit.gavita@gmail.com>
Date: Tue, 7 Jul 2026 17:23:27 +0530
Subject: [PATCH 1/2] Update package dependencies and versions in
 pyproject.toml

---
 pyproject.toml | 34 ++++++++++++++++++++++++++++------
 1 file changed, 28 insertions(+), 6 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index cf63331c..fb05733b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,7 +16,7 @@ dependencies = [
   "neuraltrain==0.0.2",
   "exca==0.5.20",
   "torch>=2.5.1,<2.7",
-  "numpy==2.2.6",
+  "numpy>=2.2.6,<3.0",
   "torchvision>=0.20,<0.22",
   "x_transformers==1.27.20",
   "einops",
@@ -27,10 +27,13 @@ dependencies = [
   "langdetect",
   "spacy",
   "soundfile",
-  "pip",
   "Levenshtein",
   "julius",
-  "transformers"
+  "transformers>=4.40.0",
+  "pandas>=2.2.0",
+  "pydantic>=2.5.0",
+  "requests>=2.31.0",
+  "tqdm>=4.66.0",
 ]
 
 [project.urls]
@@ -50,12 +53,31 @@ Repository = "https://github.com/facebookresearch/tribev2"
   ]
   training = [
     "nibabel",
-    "torchmetrics",
-    "wandb",
-    "lightning",  
+    "torchmetrics>=1.0.0",
+    "wandb>=0.16.0",
+    "lightning>=2.5.0",
   ]
   test = [
     "pytest",
+    "pytest-cov",
+  ]
+  transcription = [
+    "uvx",
+  ]
+  datasets = [
+    "datalad>=0.19.5",
+    "openneuro-py",
+    "nltk==3.8.1",
+    "nltk_contrib @ git+https://github.com/nltk/nltk_contrib.git@683961c53f0c122b90fe2d039fe795e0a2b3e997",
+  ]
+  demo = [
+    "gtts",
+    "langdetect",
+    "spacy",
+    "soundfile",
+  ]
+  all = [
+    "tribev2[plotting,training,test,transcription,datasets,demo]",
   ]
 
 [tool.black]

From 002e7ecb6b6754d9db9798cda58fe99ed965fc4d Mon Sep 17 00:00:00 2001
From: harshitgavita-07 <harshit.gavita@gmail.com>
Date: Tue, 7 Jul 2026 17:24:01 +0530
Subject: [PATCH 2/2] Add initial grid configurations and imports

Initialize grid configurations for TRIBE v2 experiments.
---
 tribev2/grids/__init__.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/tribev2/grids/__init__.py b/tribev2/grids/__init__.py
index e69de29b..d6897cd1 100644
--- a/tribev2/grids/__init__.py
+++ b/tribev2/grids/__init__.py
@@ -0,0 +1,12 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+"""Grid configurations and runners for TRIBE v2 experiments."""
+
+from .configs import base_config, mini_config
+from .defaults import default_config
+
+__all__ = ["base_config", "mini_config", "default_config"]