added stress test decorator

Author: TumanyanV

tests/test_ml_funcs.py

@@ -4,6 +4,7 @@
 from .common import upload_project
 import pytest
 
+
 sa.init(Path.home() / ".superannotate" / "config.json")
 PROJECT_NAME_VECTOR = 'ML Functionality Test Vector'
 PROJECT_NAME_PIXEL = 'ML Functionality Test Pixel'
@@ -13,6 +14,10 @@
 
 MODEL_NAME = 'Instance segmentation (trained on COCO)'
 
+@pytest.mark.skipif(
+    "SA_STRESS_TESTS" not in os.environ,
+    reason="Requires env variable to be set"
+)
 def test_run_prediction():
 
     upload_project(
@@ -51,6 +56,10 @@ def test_run_prediction():
 
 
 
+@pytest.mark.skipif(
+    "SA_STRESS_TESTS" not in os.environ,
+    reason="Requires env variable to be set"
+)
 def test_run_segmentation():
 
     model_auto = 'autonomous'