testing fix

.github/workflows/stress-test.yml

@@ -2,6 +2,8 @@ name: Run stress tests manually via the GitHub Actions UI
 
 on:
   workflow_dispatch:
+  push:
+    branches: [fix_stress_tests]
 
 permissions:
   id-token: write

integration_tests/client/datatype/test_data_generation.py

@@ -118,7 +118,9 @@ def test_generate_prediction_data(client: Client):
                 "iou_thresholds_to_keep": [0.0, 1.0],
             },
             "filters": {
-                **asdict(Filter()), # default filter properties with overrides below
+                **asdict(
+                    Filter()
+                ),  # default filter properties with overrides below
                 "annotation_types": ["box"],
                 "label_keys": ["k1"],
             },

integration_tests/stress_test.py

@@ -1,13 +1,17 @@
 # NOTE: These tests aren't run automatically on each commit. They are intended to be manually kicked-off using the [GitHub UI](https://leonardomontini.dev/github-action-manual-trigger/)
-from integration_tests.datatype.test_data_generation import (
+from integration_tests.client.datatype.test_data_generation import (
     test_generate_segmentation_data,
 )
 
-from velour.client import Client
+from velour import Client
 
 
 def test_large_dataset_upload(client: Client):
     """Tests the upload of a large dataset to velour (runtime: ~20 minutes)"""
     test_generate_segmentation_data(
-        client=client, n_images=1000, n_annotations=10, n_labels=2
+        client=client,
+        dataset_name="stress_test_dataset",
+        n_images=1000,
+        n_annotations=10,
+        n_labels=2,
     )