From b45b75350c5d8bc607ce7a6c0c10cf698311944c Mon Sep 17 00:00:00 2001
From: Maxime Jublou <jubloum@gmail.com>
Date: Mon, 18 Mar 2024 17:37:10 +0100
Subject: [PATCH] fix: Disable HuggingFace driver

---
 naas_drivers/__init__.py | 8 ++++----
 setup.py                 | 2 +-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/naas_drivers/__init__.py b/naas_drivers/__init__.py
index a45a8f40..8793d182 100644
--- a/naas_drivers/__init__.py
+++ b/naas_drivers/__init__.py
@@ -408,11 +408,11 @@ def sharepoint():
     return Sharepoint()
 
 
-@load_driver(extra_requires="ml")
-def huggingface():
-    from naas_drivers.tools.huggingface import Huggingface
+# @load_driver(extra_requires="ml")
+# def huggingface():
+#     from naas_drivers.tools.huggingface import Huggingface
 
-    return Huggingface()
+#     return Huggingface()
 
 
 @load_driver(extra_requires="twitter")
diff --git a/setup.py b/setup.py
index 213c6fc1..4dae75c1 100644
--- a/setup.py
+++ b/setup.py
@@ -91,7 +91,7 @@
     "pydash": [
         "pydash==7.0.7",
     ],
-    "ml": ["transformers==4.12.5", "tensorflow==2.6.0", "torch==1.8.1", "keras==2.6.0"],
+    # "ml": ["transformers==4.12.5", "tensorflow==2.6.0", "torch==1.8.1", "keras==2.6.0"],
     "sharepoint": ["SharePlum==0.5.1"],
     "snowflake": [
         "snowflake-connector-python==2.7.8"