sdks/python: fix linting issues

Author: mohamedawnallah

sdks/python/apache_beam/ml/rag/enrichment/milvus_search.py

@@ -584,15 +584,3 @@ def batch_elements_kwargs(self) -> Dict[str, int]:
 def join_fn(left: Embedding, right: Dict[str, Any]) -> Embedding:
   left.metadata['enrichment_data'] = right
   return left
-
-
-def unpack_dataclass_with_kwargs(dataclass_instance):
-  # Create a copy of the dataclass's __dict__.
-  params_dict: dict = dataclass_instance.__dict__.copy()
-
-  # Extract the nested kwargs dictionary.
-  nested_kwargs = params_dict.pop('kwargs', {})
-
-  # Merge the dictionaries, with nested_kwargs taking precedence
-  # in case of duplicate keys.
-  return {**params_dict, **nested_kwargs}

sdks/python/apache_beam/ml/rag/utils.py

@@ -28,6 +28,7 @@
 from typing import List
 from typing import Optional
 from typing import Tuple
+from typing import Type
 
 from apache_beam.ml.rag.types import Chunk
 from apache_beam.ml.rag.types import Content
@@ -148,7 +149,7 @@ def retry_with_backoff(
     retry_delay: float = 1.0,
     retry_backoff_factor: float = 2.0,
     operation_name: str = "operation",
-    exception_types: Tuple[type, ...] = (Exception, )
+    exception_types: Tuple[Type[BaseException], ...] = (Exception, )
 ) -> Any:
   """Executes an operation with retry logic and exponential backoff.