code: def get_table_col_names(table: str) -> Set[str]:
      with db_cursor() as cur:
          cur.execute(
              """
              SELECT column_name FROM information_schema.columns WHERE table_name = %s
              """,
              (table,),
          )
          rows = cur.fetchall()
          return {row[0] for row in rows}
  
file_path: determined/tools/scripts/replicate-metrics.py
imports: import concurrent.futures
  import contextlib
  import os
  import time
  from typing import Any, Dict, Generator, List, Set, Tuple, Union
  import psycopg
  from psycopg import sql
label_desc: Use image node_id 2 for calling a global function with example usage: get_table_col_names(table) and returns: unknown