diff --git a/cumulusci/tasks/bulkdata/mapping_parser.py b/cumulusci/tasks/bulkdata/mapping_parser.py index e630d564c6..59c7d630a2 100644 --- a/cumulusci/tasks/bulkdata/mapping_parser.py +++ b/cumulusci/tasks/bulkdata/mapping_parser.py @@ -91,9 +91,9 @@ class MappingStep(CCIDictModel): batch_size: int = None oid_as_pk: bool = False # this one should be discussed and probably deprecated record_type: Optional[str] = None # should be discussed and probably deprecated - bulk_mode: Optional[Literal["Serial", "Parallel"]] = ( - None # default should come from task options - ) + bulk_mode: Optional[ + Literal["Serial", "Parallel"] + ] = None # default should come from task options anchor_date: Optional[Union[str, date]] = None soql_filter: Optional[str] = None # soql_filter property select_options: Optional[SelectOptions] = Field( diff --git a/cumulusci/tasks/bulkdata/step.py b/cumulusci/tasks/bulkdata/step.py index b88fa8b100..b2a13bf966 100644 --- a/cumulusci/tasks/bulkdata/step.py +++ b/cumulusci/tasks/bulkdata/step.py @@ -883,30 +883,33 @@ def select_records(self, records): limit_clause = self._determine_limit_clause(total_num_records) # Generate the SOQL query based on the selection strategy - select_query, query_fields = ( - self.select_operation_executor.select_generate_query( - sobject=self.sobject, - fields=self.fields, - user_filter=self.selection_filter or None, - limit=limit_clause, - offset=None, - ) + ( + select_query, + query_fields, + ) = self.select_operation_executor.select_generate_query( + sobject=self.sobject, + fields=self.fields, + user_filter=self.selection_filter or None, + limit=limit_clause, + offset=None, ) # Execute the query and gather the records query_records = self._execute_soql_query(select_query, query_fields) # Post-process the query results for this batch - selected_records, insert_records, error_message = ( - self.select_operation_executor.select_post_process( - load_records=records, - query_records=query_records, - fields=self.fields, - num_records=total_num_records, - sobject=self.sobject, - weights=self.weights, - threshold=self.threshold, - ) + ( + selected_records, + insert_records, + error_message, + ) = self.select_operation_executor.select_post_process( + load_records=records, + query_records=query_records, + fields=self.fields, + num_records=total_num_records, + sobject=self.sobject, + weights=self.weights, + threshold=self.threshold, ) # Log the number of selected and prepared for insertion records