Merge pull request #90 from scaleapi/da-fix-tests-and-add-nightly

Fix tests and add nightly builds

Author: ardila

.circleci/config.yml

@@ -1,6 +1,7 @@
 # CircleCI jobs are only enabled to on Pull Requests and commits to master branch.
 # "Only build pull requests" enabled in Project's Advanced Settings.
 version: 2.1
+
 jobs:
   build_test:
     docker:
@@ -38,15 +39,13 @@ jobs:
             poetry run coverage run --include=nucleus/* -m pytest -s -v --junitxml=test_results/junit.xml
             poetry run coverage report
             poetry run coverage html
-
       - store_test_results:
           path: htmlcov
-
       - store_test_results:
           path: test_results
-      
       - store_artifacts:
-          path: test_results
+          path: test_results      
+
   pypi_publish:
     docker:
       - image: cimg/python:3.6
@@ -83,6 +82,16 @@ jobs:
             fi
             poetry publish --username=$PYPI_USERNAME --password=$PYPI_PASSWORD
 workflows:
+  nightly_build_test:
+    triggers:
+      - schedule:
+          cron: "0 0 * * *"
+          filters:
+            branches:
+              only:
+                - master
+    jobs:
+      - build_test
   build_test_publish:
     jobs:
       - build_test:

tests/test_prediction.py

@@ -293,7 +293,7 @@ def test_mixed_pred_upload_async(model_run: ModelRun):
         "job_id": job.id,
         "status": "Completed",
         "message": {
-            "annotation_upload": {
+            "prediction_upload": {
                 "epoch": 1,
                 "total": 2,
                 "errored": 0,
@@ -331,7 +331,7 @@ def test_mixed_pred_upload_async_with_error(model_run: ModelRun):
         "job_id": job.id,
         "status": "Completed",
         "message": {
-            "annotation_upload": {
+            "prediction_upload": {
                 "epoch": 1,
                 "total": 2,
                 "errored": 1,