Added check for predictions file

.github/workflows/checks.yaml

@@ -41,6 +41,13 @@ jobs:
       - name: Run prediction
         run: docker run --rm -v "$(pwd)/.:/data" eyra-rank:latest /data/PreFer_fake_data.csv /data/PreFer_fake_background_data.csv --output /data/predictions.csv
 
+      - name: Check if file exists
+        run: |
+          if [ ! -f "predictions.csv" ]; then
+            echo "Predictions file not found. Please check the logs to see what went wrong."
+            exit 1
+          fi
+
       - name: Build Docker scoring image
         uses: docker/build-push-action@v4
         with:

run.py

@@ -66,10 +66,5 @@ def predict(data_path, background_data_path, output):
 
 
 if __name__ == "__main__":
-    import sys
-
-    print("Forced error")
-    sys.exit(1)
-
     args = parser.parse_args()
     predict(args.data_path, args.background_data_path, args.output)