Improve table_exists() in example_merlion.py

crate · Sep 18, 2023 · e589f44 · e589f44
1 parent 80f9d6f
commit e589f44
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/CHANGES.md b/CHANGES.md
@@ -3,6 +3,7 @@
 
 ## in progress
 - Update to MLflow 2.7.1
+- Improve `table_exists()` in `example_merlion.py`
 
 ## 2023-09-12 0.1.1
 - Documentation: Improve "Container Usage" page

diff --git a/examples/tracking_merlion.py b/examples/tracking_merlion.py
@@ -47,15 +47,15 @@ def connect_database():
     return client.connect(dburi)
 
 
-def table_exists(table_name: str, schema_name: str = "doc") -> bool:
+def table_exists(table_name: str) -> bool:
     """
     Check if database table exists.
     """
     conn = connect_database()
     cursor = conn.cursor()
     sql = (
         f"SELECT table_name FROM information_schema.tables "  # noqa: S608
-        f"WHERE table_name = '{table_name}' AND table_schema = '{schema_name}'"
+        f"WHERE table_name = '{table_name}' AND table_schema = CURRENT_SCHEMA"
     )
     cursor.execute(sql)
     rowcount = cursor.rowcount