locust 2.41.2.dev9__tar.gz → 2.41.3.dev2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: locust
-Version: 2.41.2.dev9
+Version: 2.41.3.dev2
 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.dev2}/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.dev2'
+__version_tuple__ = version_tuple = (2, 41, 3, 'dev2')
 
 __commit_id__ = commit_id = None

{locust-2.41.2.dev9 → locust-2.41.3.dev2}/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]):