Skip to content

Benchmark db manager

write_table(conn, df, table_name)

Write table to DuckDB database. Args: conn (DuckDBPyConnection): DuckDB connection. df (pl.DataFrame): DuckDB dataframe. table_name (str): Table name.

Source code in src/pheval/analyse/benchmark_db_manager.py 
14
15
16
17
18
19
20
21
22
23
def write_table(conn: DuckDBPyConnection, df: pl.DataFrame, table_name: str) -> None:
    """
    Write table to DuckDB database.
    Args:
        conn (DuckDBPyConnection): DuckDB connection.
        df (pl.DataFrame): DuckDB dataframe.
        table_name (str): Table name.
    """
    logger.info(f"Storing results in {table_name}.")
    conn.execute(f"""CREATE TABLE "{table_name}" AS SELECT * FROM df""")