pylint fix take 3

Author: verayu43

src/sagemaker/feature_store/feature_group.py

@@ -270,19 +270,24 @@ def _run_multi_process(self, data_frame: DataFrame, wait=True, timeout=None):
             timeout (Union[int, float]): ``concurrent.futures.TimeoutError`` will be raised
                 if timeout is reached.
         """
+        # pylint: disable=I1101
         batch_size = math.ceil(data_frame.shape[0] / self.max_processes)
+        # pylint: enable=I1101
 
         args = []
         for i in range(self.max_processes):
             start_index = min(i * batch_size, data_frame.shape[0])
             end_index = min(i * batch_size + batch_size, data_frame.shape[0])
-            args += [(
-                self.max_workers,
-                self.feature_group_name,
-                self.sagemaker_fs_runtime_client_config,
-                data_frame[start_index:end_index],
-                start_index,
-                timeout)]
+            args += [
+                (
+                    self.max_workers,
+                    self.feature_group_name,
+                    self.sagemaker_fs_runtime_client_config,
+                    data_frame[start_index:end_index],
+                    start_index,
+                    timeout,
+                )
+            ]
 
         def init_worker():
             # ignore keyboard interrupts in child processes.
@@ -298,10 +303,14 @@ def init_worker():
             self.wait(timeout=timeout)
 
     def _run_multi_threaded(
-            self,
-            max_workers: int, feature_group_name: str,
-            sagemaker_fs_runtime_client_config: Config,
-            data_frame: DataFrame, row_offset=0, timeout=None) -> List[int]:
+        self,
+        max_workers: int,
+        feature_group_name: str,
+        sagemaker_fs_runtime_client_config: Config,
+        data_frame: DataFrame,
+        row_offset=0,
+        timeout=None,
+    ) -> List[int]:
         """Start the ingestion process.
 
         Args:
@@ -316,7 +325,9 @@ def _run_multi_threaded(
             List of row indices that failed to be ingested.
         """
         executor = ThreadPoolExecutor(max_workers=max_workers)
+        # pylint: disable=I1101
         batch_size = math.ceil(data_frame.shape[0] / max_workers)
+        # pylint: enable=I1101
 
         futures = {}
         for i in range(max_workers):