lints

Author: claudevdm

sdks/python/apache_beam/ml/rag/ingestion/spanner.py

@@ -65,7 +65,6 @@
 from dataclasses import dataclass
 from typing import Any
 from typing import Callable
-from typing import Dict
 from typing import List
 from typing import Literal
 from typing import NamedTuple
@@ -291,7 +290,7 @@ def default_fn(chunk: Chunk) -> str:
   def with_metadata_spec(
       self,
       column_name: str = "metadata",
-      value_fn: Optional[Callable[[Chunk], Dict[str, Any]]] = None
+      value_fn: Optional[Callable[[Chunk], Any]] = None
   ) -> 'SpannerColumnSpecsBuilder':
     """Add metadata JSON column.
     

sdks/python/apache_beam/ml/rag/ingestion/spanner_it_test.py

@@ -17,6 +17,7 @@
 
 """Integration tests for Spanner vector store writer."""
 
+import logging
 import os
 import time
 import unittest
@@ -211,7 +212,6 @@ def test_write_default_schema(self):
     self.assertEqual(row2[0], 'doc2')
     self.assertEqual(list(row2[1]), [4.0, 5.0, 6.0])
     self.assertEqual(row2[2], 'Second document')
-    logging.warning(f"CLAUDE {results}")
 
   def test_write_flattened_metadata(self):
     """Test writing with flattened metadata fields."""
@@ -594,7 +594,6 @@ def test_write_with_timestamp(self):
     self.assertEqual(rows[0][0], 'doc1')
     # Timestamp is returned as datetime object by Spanner client
     self.assertIsNotNone(rows[0][1])
-    logging.warning(rows)
 
 
 if __name__ == '__main__':