Update bert-sklearn package (#139)

scikit-learn-contrib · Dec 6, 2024 · dfe47d1 · dfe47d1
1 parent 44d8e0f
commit dfe47d1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.github/workflows/test-pr.yml b/.github/workflows/test-pr.yml
@@ -3,11 +3,13 @@ name: Test PR
 on: [pull_request]
 
 jobs:
+
   lint:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v2
       - uses: psf/black@stable
+
   test:
     runs-on: ${{ matrix.os }}
     strategy:

diff --git a/setup.py b/setup.py
@@ -45,7 +45,7 @@
         "ray",
         "shap==0.44.1",
         "xarray==2023.1.0",
-        "bert-sklearn @ git+https://github.com/charles9n/bert-sklearn.git#egg=bert-sklearn",
+        "bert-sklearn",
     ],
 }