Format code with black

Author: dmsmith25

vsb/cmdline_args.py

@@ -525,7 +525,10 @@ def validate_parsed_args(
                     parser.error(
                         "multi_namespace mode requires --skip_populate. Multi-namespace benchmarking only works with existing populated indexes."
                     )
-                if getattr(args, "pinecone_namespace_name", "__default__") != "__default__":
+                if (
+                    getattr(args, "pinecone_namespace_name", "__default__")
+                    != "__default__"
+                ):
                     parser.error(
                         "multi_namespace mode does not support custom namespace names. Use --pinecone_multi_namespace to auto-discover all namespaces."
                     )

vsb/databases/pinecone/pinecone.py

@@ -199,7 +199,7 @@ def __init__(
                     f"PineconeDB: Index '{self.index_name}' does not exist. Multi-namespace mode requires an existing populated index. Please create the index first or use single-namespace mode."
                 )
                 raise StopUser()
-            
+
             logger.info(
                 f"PineconeDB: Specified index '{self.index_name}' was not found, or the "
                 f"specified API key cannot access it. Creating new index '{self.index_name}'."
@@ -242,22 +242,22 @@ def __init__(
 
         # Initialize namespaces list (empty for both modes to avoid AttributeError)
         self.namespaces = []
-        
+
         # Multi-namespace mode: discover and validate namespaces
         if self.multi_namespace:
             if self.created_index:
                 logger.critical(
                     f"PineconeDB: Cannot use multi_namespace mode with a newly created index. Index '{self.index_name}' must already exist and be populated."
                 )
                 raise StopUser()
-            
+
             self.namespaces = self._discover_all_namespaces()
             if not self.namespaces:
                 logger.critical(
                     f"PineconeDB: No populated namespaces found in index '{self.index_name}'. Multi-namespace mode requires at least one namespace with records."
                 )
                 raise StopUser()
-            
+
             logger.info(
                 f"PineconeDB: Discovered {len(self.namespaces)} namespaces: {', '.join(self.namespaces)}"
             )
@@ -381,12 +381,12 @@ def _discover_all_namespaces(self) -> list[str]:
             raise ValueError(
                 f"Failed to list namespaces in index '{self.index_name}': {e}"
             ) from e
-        
+
         if not populated_namespaces:
             raise ValueError(
                 f"No populated namespaces found in index '{self.index_name}'. Multi-namespace mode requires at least one namespace with records."
             )
-        
+
         return sorted(populated_namespaces)
 
     def _get_namespaces_for_user(self, user_id: int, num_users: int) -> list[str]:
@@ -395,7 +395,9 @@ def _get_namespaces_for_user(self, user_id: int, num_users: int) -> list[str]:
             raise ValueError(
                 f"Cannot distribute {num_users} users across {len(self.namespaces)} namespaces. Number of users must be <= number of namespaces."
             )
-        return [self.namespaces[i] for i in range(user_id, len(self.namespaces), num_users)]
+        return [
+            self.namespaces[i] for i in range(user_id, len(self.namespaces), num_users)
+        ]
 
     def check_namespace_exists(self, namespace: str) -> bool:
         """Check if a namespace exists inside the current index using list_namespaces generator."""

vsb/users.py

@@ -250,9 +250,9 @@ def __init__(self, environment):
             f"Initialising RunUser id:{self.user_id}, target request/sec:{target_throughput}"
         )
         self.query_iter = None
-        
+
         # Multi-namespace mode: get namespaces assigned to this user
-        if hasattr(self.database, 'multi_namespace') and self.database.multi_namespace:
+        if hasattr(self.database, "multi_namespace") and self.database.multi_namespace:
             self.user_namespaces = self.database._get_namespaces_for_user(
                 self.user_id, self.users_total
             )
@@ -291,9 +291,13 @@ def do_run(self):
             base_iter = self.workload.get_query_iter(
                 self.users_total, self.user_id, batch_size
             )
-            
+
             # Multi-namespace mode: wrap iterator to inject namespace names
-            if hasattr(self.database, 'multi_namespace') and self.database.multi_namespace:
+            if (
+                hasattr(self.database, "multi_namespace")
+                and self.database.multi_namespace
+            ):
+
                 def namespace_wrapper():
                     for tenant, request in base_iter:
                         # Select next namespace using round-robin
@@ -302,6 +306,7 @@ def namespace_wrapper():
                         ]
                         self.namespace_index += 1
                         yield (namespace, request)
+
                 self.query_iter = namespace_wrapper()
             else:
                 # Single namespace mode: use iterator as-is