Add bert-sklearn to dev dependencies

scikit-learn-contrib · Dec 2, 2024 · 65958e6 · 65958e6
1 parent a92b6a8
commit 65958e6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -45,6 +45,7 @@
         "ray",
         "shap==0.44.1",
         "xarray==2023.1.0",
+        "bert-sklearn @ git+ssh://[email protected]/charles9n/bert-sklearn.git#egg=bert-sklearn",
     ],
 }