Merge pull request #1086 from roboflow/fix/test_multi_label_classification_workflow_assertions_for_prod

Update E2E test assertions for test_multi_label_classification_workflow

Author: grzegorz-roboflow

tests/inference/hosted_platform_tests/workflows_examples/roboflow_models/v1/test_workflow_for_classification.py

@@ -122,8 +122,10 @@ def test_multi_class_classification_workflow(
         {"dog"},
         {"cat", "dog"},
     ],
-    PlatformEnvironment.ROBOFLOW_PLATFORM: [{"cat", "dog"}, {"cat", "dog"}],
+    PlatformEnvironment.ROBOFLOW_PLATFORM: [{"dog"}, set()],
 }
+MULTI_LABEL_CLASSIFICATION_RESULTS_FOR_ENVIRONMENT[PlatformEnvironment.ROBOFLOW_STAGING_LOCALHOST] = MULTI_LABEL_CLASSIFICATION_RESULTS_FOR_ENVIRONMENT[PlatformEnvironment.ROBOFLOW_STAGING]
+MULTI_LABEL_CLASSIFICATION_RESULTS_FOR_ENVIRONMENT[PlatformEnvironment.ROBOFLOW_PLATFORM_LOCALHOST] = MULTI_LABEL_CLASSIFICATION_RESULTS_FOR_ENVIRONMENT[PlatformEnvironment.ROBOFLOW_PLATFORM]
 
 
 @pytest.mark.flaky(retries=4, delay=1)

tests/inference/hosted_platform_tests/workflows_examples/roboflow_models/v2/test_workflow_for_classification.py

@@ -40,7 +40,8 @@
     PlatformEnvironment.ROBOFLOW_STAGING: [0.3667, 0.5917],
     PlatformEnvironment.ROBOFLOW_PLATFORM: [0.8252, 0.9962],
 }
-
+MULTI_CLASS_CLASSIFICATION_RESULTS_FOR_ENVIRONMENT[PlatformEnvironment.ROBOFLOW_STAGING_LOCALHOST] = MULTI_CLASS_CLASSIFICATION_RESULTS_FOR_ENVIRONMENT[PlatformEnvironment.ROBOFLOW_STAGING]
+MULTI_CLASS_CLASSIFICATION_RESULTS_FOR_ENVIRONMENT[PlatformEnvironment.ROBOFLOW_PLATFORM_LOCALHOST] = MULTI_CLASS_CLASSIFICATION_RESULTS_FOR_ENVIRONMENT[PlatformEnvironment.ROBOFLOW_PLATFORM]
 
 @pytest.mark.flaky(retries=4, delay=1)
 def test_multi_class_classification_workflow(
@@ -122,8 +123,10 @@ def test_multi_class_classification_workflow(
         {"dog"},
         {"cat", "dog"},
     ],
-    PlatformEnvironment.ROBOFLOW_PLATFORM: [{"cat", "dog"}, {"cat", "dog"}],
+    PlatformEnvironment.ROBOFLOW_PLATFORM: [{"dog"}, set()],
 }
+MULTI_LABEL_CLASSIFICATION_RESULTS_FOR_ENVIRONMENT[PlatformEnvironment.ROBOFLOW_STAGING_LOCALHOST] = MULTI_LABEL_CLASSIFICATION_RESULTS_FOR_ENVIRONMENT[PlatformEnvironment.ROBOFLOW_STAGING]
+MULTI_LABEL_CLASSIFICATION_RESULTS_FOR_ENVIRONMENT[PlatformEnvironment.ROBOFLOW_PLATFORM_LOCALHOST] = MULTI_LABEL_CLASSIFICATION_RESULTS_FOR_ENVIRONMENT[PlatformEnvironment.ROBOFLOW_PLATFORM]
 
 
 @pytest.mark.flaky(retries=4, delay=1)