lint + test update

graphistry · Apr 12, 2024 · 0ba0a1d · 0ba0a1d
1 parent 7080061
commit 0ba0a1d
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 4 deletions.
diff --git a/.github/workflows/on_push.yml b/.github/workflows/on_push.yml
@@ -30,14 +30,27 @@ jobs:
           python -m pip install --upgrade pip
           python -m pip install -e .
 
-    - name: Minimal tests
+    - name: gpy_feat tests
+      run: |
+          source dots/bin/activate
+          python DOTS/test/test_dots_feat.py -n 100 -e 1 -d 0 -o dots_gpy_feats.csv
+      env:
+          OS_TOKEN: ${{ secrets.OS_TOKEN }}
+
+    - name: distilroberta tests
       run: |
           source dots/bin/activate
           python -m spacy download en_core_web_sm
-          python DOTS/test/test_dots_feat.py -n 5 -f 3 -d 4 -o dots_feats.csv
+          python DOTS/test/test_dots_feat.py -n 100 -e 0 -d 0 -o dots_drba_feats.csv
+      env:
+          OS_TOKEN: ${{ secrets.OS_TOKEN }}
+
+    - name: gliner tests
+      run: |
+          source dots/bin/activate
+          python DOTS/test/test_dots_feat.py -n 100 -e 2 -d 0 -o dots_glnr_feats.csv
       env:
           OS_TOKEN: ${{ secrets.OS_TOKEN }}
-          LOBSTR_KEY: ${{ secrets.LOBSTR_KEY }}
 
     # - name: Minimal tests 2
     #   run: |

diff --git a/setup.py b/setup.py
@@ -22,7 +22,7 @@
   'validators',
   'pytesseract',
   'selenium',
-  'webdriver_manager,'
+  'webdriver_manager',
   'undetected_chromedriver',
   'gliner',