Revert "Benchmark time reduce (#396)"

This reverts commit f4e4cf4.
deeppavlov · Oct 18, 2024 · bc3bd91 · bc3bd91
1 parent f4e4cf4
commit bc3bd91
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/chatsky/utils/db_benchmark/basic_config.py b/chatsky/utils/db_benchmark/basic_config.py
@@ -96,9 +96,9 @@ class BasicBenchmarkConfig(BenchmarkConfig, frozen=True):
     Number of times the contexts will be benchmarked.
     Increasing this number decreases standard error of the mean for benchmarked data.
     """
-    from_dialog_len: int = 25
+    from_dialog_len: int = 300
     """Starting dialog len of a context."""
-    to_dialog_len: int = 50
+    to_dialog_len: int = 311
     """
     Final dialog len of a context.
     :py:meth:`~.BasicBenchmarkConfig.context_updater` will return contexts
@@ -177,25 +177,27 @@ def context_updater(self, context: Context) -> Optional[Context]:
 basic_configurations = {
     "large-misc": BasicBenchmarkConfig(
         from_dialog_len=1,
-        to_dialog_len=26,
+        to_dialog_len=50,
         message_dimensions=(3, 5, 6, 5, 3),
         misc_dimensions=(2, 4, 3, 8, 100),
     ),
     "short-messages": BasicBenchmarkConfig(
+        from_dialog_len=500,
+        to_dialog_len=550,
         message_dimensions=(2, 30),
         misc_dimensions=(0, 0),
     ),
     "default": BasicBenchmarkConfig(),
-    "large-misc-long-dialog": BasicBenchmarkConfig(
-        from_dialog_len=50,
-        to_dialog_len=75,
+    "large-misc--long-dialog": BasicBenchmarkConfig(
+        from_dialog_len=500,
+        to_dialog_len=550,
         message_dimensions=(3, 5, 6, 5, 3),
         misc_dimensions=(2, 4, 3, 8, 100),
     ),
     "very-long-dialog-len": BasicBenchmarkConfig(
         context_num=10,
-        from_dialog_len=1000,
-        to_dialog_len=1050,
+        from_dialog_len=10000,
+        to_dialog_len=10050,
     ),
     "very-long-message-len": BasicBenchmarkConfig(
         context_num=10,