locust 2.41.2.dev9__tar.gz → 2.41.3.dev4__tar.gz

{locust-2.41.2.dev9 → locust-2.41.3.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: locust
-Version: 2.41.2.dev9
+Version: 2.41.3.dev4
 Summary: Developer-friendly load testing framework
 Project-URL: homepage, https://locust.io/
 Project-URL: repository, https://github.com/locustio/locust

{locust-2.41.2.dev9 → locust-2.41.3.dev4}/locust/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '2.41.2.dev9'
-__version_tuple__ = version_tuple = (2, 41, 2, 'dev9')
+__version__ = version = '2.41.3.dev4'
+__version_tuple__ = version_tuple = (2, 41, 3, 'dev4')
 
 __commit_id__ = commit_id = None

{locust-2.41.2.dev9 → locust-2.41.3.dev4}/locust/contrib/milvus.py

@@ -21,7 +21,7 @@ class BaseClient(ABC):
         pass
 
     @abstractmethod
-    def create_collection(self, schema, index_params) -> None:
+    def create_collection(self, schema, index_params, **kwargs) -> None:
         pass
 
     @abstractmethod
@@ -80,11 +80,12 @@ class MilvusV2Client(BaseClient):
     def close(self):
         self.client.close()
 
-    def create_collection(self, schema, index_params):
+    def create_collection(self, schema, index_params, **kwargs):
         self.client.create_collection(
             collection_name=self.collection_name,
             schema=schema,
             index_params=index_params,
+            **kwargs,
         )
 
     def insert(self, data):
@@ -299,7 +300,7 @@ class MilvusUser(User):
         timeout: int = 60,
         schema: CollectionSchema | None = None,
         index_params: IndexParams | None = None,
-        **client_kwargs,
+        **kwargs,  # enable_dynamic_field, num_shards, consistency_level etc. ref: https://milvus.io/api-reference/pymilvus/v2.6.x/MilvusClient/Collections/create_collection.md
     ):
         super().__init__(environment)
 
@@ -317,7 +318,7 @@ class MilvusUser(User):
             timeout=timeout,
         )
         if schema is not None:
-            self.client.create_collection(schema=schema, index_params=index_params)
+            self.client.create_collection(schema=schema, index_params=index_params, **kwargs)
 
     @staticmethod
     def _fire_event(request_type: str, name: str, result: dict[str, Any]):

{locust-2.41.2.dev9 → locust-2.41.3.dev4}/locust/main.py

@@ -584,11 +584,11 @@ See https://github.com/locustio/locust/wiki/Installation#increasing-maximum-numb
                 headless_master_greenlet = gevent.spawn(runner.start, options.num_users, options.spawn_rate)
                 headless_master_greenlet.link_exception(greenlet_exception_handler)
 
-        if options.run_time:
-            logger.info(f"Run time limit set to {options.run_time} seconds")
-            spawn_run_time_quit_greenlet()
-        elif not options.worker and not environment.shape_class:
-            logger.info("No run time limit set, use CTRL+C to interrupt")
+            if options.run_time:
+                logger.info(f"Run time limit set to {options.run_time} seconds")
+                spawn_run_time_quit_greenlet()
+            elif not environment.shape_class:
+                logger.info("No run time limit set, use CTRL+C to interrupt")
 
     if options.csv_prefix:
         gevent.spawn(stats_csv_writer.stats_writer).link_exception(greenlet_exception_handler)