wip

Author: DaltheCow

tests/integration/executor/test_report_generation.py

@@ -16,10 +16,10 @@ def test_executor_openai_single_report_generation_sync_mode(
     request_genrator = dummy.services.TestRequestGenerator(
         tokenizer="bert-base-uncased"
     )
-    profile_generation_mode = ProfileGenerationMode.SINGLE
+    profile_generation_mode = ProfileGenerationMode.FIXED_RATE
     profile_generator_kwargs = {
         "rate_type": LoadGenerationMode.SYNCHRONOUS,
-        "rate": 1.0,
+        "rate": [1.0],
     }
 
     executor = Executor(
@@ -55,10 +55,10 @@ def test_executor_openai_single_report_generation_constant_mode_infinite(
     request_genrator = dummy.services.TestRequestGenerator(
         tokenizer="bert-base-uncased"
     )
-    profile_generation_mode = ProfileGenerationMode.SINGLE
+    profile_generation_mode = ProfileGenerationMode.FIXED_RATE
     profile_generator_kwargs = {
         "rate_type": LoadGenerationMode.CONSTANT,
-        "rate": 1.0,
+        "rate": [1.0],
     }
 
     executor = Executor(
@@ -88,10 +88,10 @@ def test_executor_openai_single_report_generation_constant_mode_limited(
     request_genrator = dummy.services.TestRequestGenerator(
         tokenizer="bert-base-uncased"
     )
-    profile_generation_mode = ProfileGenerationMode.SINGLE
+    profile_generation_mode = ProfileGenerationMode.FIXED_RATE
     profile_generator_kwargs = {
         "rate_type": LoadGenerationMode.CONSTANT,
-        "rate": 1.0,
+        "rate": [1.0],
     }
 
     executor = Executor(
@@ -124,10 +124,10 @@ def test_executor_openai_single_report_generation_constant_mode_failed(
     request_genrator = dummy.services.TestRequestGenerator(
         tokenizer="bert-base-uncased"
     )
-    profile_generation_mode = ProfileGenerationMode.SINGLE
+    profile_generation_mode = ProfileGenerationMode.FIXED_RATE
     profile_generator_kwargs = {
         "rate_type": LoadGenerationMode.CONSTANT,
-        "rate": 1.0,
+        "rate": [1.0],
     }
 
     executor = Executor(
@@ -153,10 +153,10 @@ def test_executor_openai_single_report_generation_constant_mode_cancelled_report
     request_genrator = dummy.services.TestRequestGenerator(
         tokenizer="bert-base-uncased"
     )
-    profile_generation_mode = ProfileGenerationMode.SINGLE
+    profile_generation_mode = ProfileGenerationMode.FIXED_RATE
     profile_generator_kwargs = {
         "rate_type": LoadGenerationMode.CONSTANT,
-        "rate": 1.0,
+        "rate": [1.0],
     }
 
     executor = Executor(