dbnl.util

get_column_schemas_from_dataframe

dbnl.util.get_column_schemas_from_dataframe(df: DataFrame) → list[RunSchemaColumnSchemaDict]

get_default_components_dag_from_column_schemas

dbnl.util.get_default_components_dag_from_column_schemas(column_schemas: Sequence[ColumnSchemaDict]) → dict[str, list[str]] | None

Gets the unconnected components DAG from a list of column schemas. If there are no components, returns None. The default components dag is of the form {

“component1”: [], “component2”: [], …}

• Parameters:column_schemas – list of column schemas

• Returns: dictionary of components DAG or None

get_run_config_column_schemas_from_dataframe

dbnl.util.get_run_config_column_schemas_from_dataframe(df: DataFrame) → list[RunConfigPrimitiveColumnSchemaDict | RunConfigContainerColumnSchemaDict]

get_run_config_scalar_schemas_from_dataframe

dbnl.util.get_run_config_scalar_schemas_from_dataframe(df: DataFrame) → list[RunConfigPrimitiveScalarSchemaDict | RunConfigContainerScalarSchemaDict]

get_run_schema_columns_from_dataframe

dbnl.util.get_run_schema_columns_from_dataframe(df: DataFrame) → list[RunSchemaColumnSchema]

get_run_schema_scalars_from_dataframe

dbnl.util.get_run_schema_scalars_from_dataframe(df: DataFrame) → list[RunSchemaScalarSchema]

get_scalar_schemas_from_dataframe

dbnl.util.get_scalar_schemas_from_dataframe(df: DataFrame) → list[RunSchemaScalarSchemaDict]

make_test_session_input

dbnl.util.make_test_session_input(*, run: Run | None = None, run_query: RunQuery | None = None, run_alias: str = 'EXPERIMENT') → TestSessionInput

Create a TestSessionInput object from a Run or a RunQuery. Useful for creating TestSessions right after closing a Run.

• Parameters:

  • run – The Run to create the TestSessionInput from

  • run_query – The RunQuery to create the TestSessionInput from

  • run_alias – Alias for the Run, must be ‘EXPERIMENT’ or ‘BASELINE’, defaults to “EXPERIMENT”

• Raises:DBNLInputValidationError – If both run and run_query are None

• Returns: TestSessionInput object