sdks/python: remove SQL_TABLE_ID env variable

Author: mohamedawnallah

sdks/python/apache_beam/examples/snippets/transforms/elementwise/enrichment.py

@@ -64,7 +64,7 @@ def enrichment_with_cloudsql():
   database_user = os.environ.get("SQL_DB_USER")
   database_password = os.environ.get("SQL_DB_PASSWORD")
   database_id = os.environ.get("SQL_DB_ID")
-  table_id = os.environ.get("SQL_TABLE_ID")
+  table_id = "products"
   where_clause_template = "product_id = {}"
   where_clause_fields = ["product_id"]
 

sdks/python/apache_beam/examples/snippets/transforms/elementwise/enrichment_test.py

@@ -114,6 +114,7 @@ def test_enrichment_with_vertex_ai_legacy(self, mock_stdout):
     self.assertEqual(output, expected)
 
   def pre_cloudsql_enrichment_test(self):
+    table_id ="products"
     columns = [
         Column("product_id", Integer, primary_key=True),
         Column("name", String, nullable=False),
@@ -138,10 +139,9 @@ def pre_cloudsql_enrichment_test(self):
     os.environ['SQL_DB_PASSWORD'] = db.password
     os.environ['SQL_DB_ID'] = db.id
     os.environ['SQL_DB_URL'] = db.url
-    os.environ['SQL_TABLE_ID'] = "products"
     engine = CloudSQLEnrichmentTestHelper.create_table(
-        table_id=os.environ.get("SQL_TABLE_ID"),
         db_url=os.environ.get("SQL_DB_URL"),
+        table_id=table_id,
         columns=columns,
         table_data=table_data)
     return db, engine
@@ -156,7 +156,6 @@ def post_cloudsql_enrichment_test(
     os.environ.pop('SQL_DB_PASSWORD', None)
     os.environ.pop('SQL_DB_ID', None)
     os.environ.pop('SQL_DB_URL', None)
-    os.environ.pop('SQL_TABLE_ID', None)
 
 
 if __name__ == '__main__':