fix: Disable HuggingFace driver

jupyter-naas · Mar 18, 2024 · b45b753 · b45b753
1 parent 33a77af
commit b45b753
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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"