Improve CI

.github/workflows/python-publish.yml

@@ -15,7 +15,10 @@ jobs:
         python-version: [3.8]
 
     steps:
-      - uses: actions/checkout@v2
+      - name: Checkout code
+        uses: nschloe/action-cached-lfs-checkout@v1.1.3
+        with:
+          include: "*.0.onnx"
 
       - uses: actions/setup-python@v2
         name: Install Python
@@ -36,7 +39,10 @@ jobs:
     name: Build source distribution
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v2
+      - name: Checkout code
+        uses: nschloe/action-cached-lfs-checkout@v1.1.3
+        with:
+          include: "*.0.onnx"
 
       - uses: actions/setup-python@v2
         name: Install Python

.github/workflows/testing.yml

@@ -21,7 +21,7 @@ jobs:
       - name: Checkout code
         uses: nschloe/action-cached-lfs-checkout@v1.1.3
         with:
-          include: "*.0.onnx"
+          exclude: "scoutbot/*/models/pytorch/"
 
       - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v2

tests/test_loc.py

@@ -92,5 +92,7 @@ def test_loc_onnx_pipeline():
         for key in target.keys():
             if key == 'class_label':
                 assert getattr(output, key) == target.get(key)
+            elif key == 'confidence':
+                assert abs(getattr(output, key) - target.get(key)) < 1e-4
             else:
-                assert abs(getattr(output, key) - target.get(key)) < 1e-2
+                assert abs(getattr(output, key) - target.get(key)) < 1e-1