From dfe47d1b36e0669d13f0fb9b16dbcbb1940a16e1 Mon Sep 17 00:00:00 2001
From: mirand863 <miranda.fmm@gmail.com>
Date: Fri, 6 Dec 2024 18:23:15 -0300
Subject: [PATCH] Update bert-sklearn package (#139)

---
 .github/workflows/test-pr.yml | 2 ++
 setup.py                      | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/test-pr.yml b/.github/workflows/test-pr.yml
index 7cdb3c3b..fe18840f 100644
--- 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
index 4d80b4d8..7d674908 100644
--- 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",
     ],
 }