fix pylint

Author: verayu43

src/sagemaker/feature_store/feature_group.py

@@ -241,7 +241,7 @@ def wait(self, timeout=None):
         results = None
         try:
             results = self._async_result.get(timeout=timeout)
-        except KeyboardInterrupt or NotImplementedError as i:
+        except KeyboardInterrupt as i:
             # terminate workers abruptly on keyboard interrupt.
             self._processing_pool.terminate()
             self._processing_pool.close()
@@ -252,20 +252,18 @@ def wait(self, timeout=None):
             self._processing_pool.close()
             self._processing_pool.clear()
 
-        if results == None:
-            return
-        else:
+        if results:
             self._failed_indices = [
                 failed_index for failed_indices in results for failed_index in failed_indices
             ]
 
-        if len(self._failed_indices) > 0:
-            raise IngestionError(
-                self._failed_indices,
-                f"Failed to ingest some data into FeatureGroup {self.feature_group_name}",
-            )
+            if len(self._failed_indices) > 0:
+                raise IngestionError(
+                    self._failed_indices,
+                    f"Failed to ingest some data into FeatureGroup {self.feature_group_name}",
+                )
 
-    def _run_multi_process(self, data_frame: DataFrame, wait=True, timeout=None): # Not in use
+    def _run_multi_process(self, data_frame: DataFrame, wait=True, timeout=None):  # Not in use
         """Start the ingestion process with the specified number of processes.
 
         Args:
@@ -348,7 +346,7 @@ def run(self, data_frame: DataFrame, wait=True, timeout=None):
             timeout (Union[int, float]): ``concurrent.futures.TimeoutError`` will be raised
                 if timeout is reached.
         """
-        #self._run_multi_process(data_frame=data_frame, wait=wait, timeout=timeout)
+        # self._run_multi_process(data_frame=data_frame, wait=wait, timeout=timeout)
         self._run_multi_threaded(data_frame=data_frame, timeout=timeout)