Skip to content

archaeo_super_prompt.dataset.postgresql_engine

source module archaeo_super_prompt.dataset.postgresql_engine

Structured data loading from a remote dataset.

This module manages the interaction with the postgresql database to load a pandas DataFrame. The sqlalchemy library with the psycopg2 engine are used.

Functions

  • get_entries — Fetch from the remote database a set of samples of interventions.

  • get_entries_with_ids — Fetch on the db the metadata of the intervention with the given ids.

source get_entries(max_number: int, seed: float, only_recent_entries=False)

Fetch from the remote database a set of samples of interventions.

source get_entries_with_ids(ids: set[int])

Fetch on the db the metadata of the intervention with the given ids.