From a11e07213966c669688efff2339b93bedc07791a Mon Sep 17 00:00:00 2001
From: Soumik Rakshit <19soumik.rakshit96@gmail.com>
Date: Tue, 29 Oct 2024 23:44:59 +0530
Subject: [PATCH] add: huggingface integration unit testing shard

---
 .github/workflows/test.yaml | 1 +
 noxfile.py                  | 1 +
 pyproject.toml              | 1 +
 3 files changed, 3 insertions(+)

diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
index 0605b0534df..c92f48f480d 100644
--- a/.github/workflows/test.yaml
+++ b/.github/workflows/test.yaml
@@ -242,6 +242,7 @@ jobs:
             'openai',
             'scorers_tests',
             'pandas-test',
+            'huggingface',
           ]
       fail-fast: false
     services:
diff --git a/noxfile.py b/noxfile.py
index 90aa3bfaac4..8ee3698f595 100644
--- a/noxfile.py
+++ b/noxfile.py
@@ -43,6 +43,7 @@ def lint(session):
         "openai",
         "scorers_tests",
         "pandas-test",
+        "huggingface",
     ],
 )
 def tests(session, shard):
diff --git a/pyproject.toml b/pyproject.toml
index 407b7548327..219343ae3f1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -72,6 +72,7 @@ notdiamond = ["notdiamond>=0.3.21", "litellm<=1.49.1"]
 openai = ["openai>=1.0.0"]
 pandas-test = ["pandas>=2.2.3"]
 modal = ["modal", "python-dotenv"]
+huggingface = ["huggingface-hub>=0.26.2"]
 test = [
   "nox",
   "pytest>=8.2.0",