kumoai.experimental.rfm.Table

kumoai.experimental.rfm.Table#

class kumoai.experimental.rfm.Table[source]#

Bases: ABC

A Table fully specifies the relevant metadata of a single table, i.e. its selected columns, data types, semantic types, primary keys and time columns.

Parameters:

  • name (str) – The name of this table.

  • source_name (Optional[str]) – The source name of this table. If set to None, name is being used.

  • columns (Optional[Sequence[Union[ColumnSpec, Mapping[str, Any], str]]]) – The selected columns of this table.

  • primary_key (MissingType | str | None) – The name of the primary key of this table, if it exists.

  • time_column (Optional[str]) – The name of the time column of this table, if it exists.

  • end_time_column (Optional[str]) – The name of the end time column of this table, if it exists.

__init__(name, source_name=None, columns=None, primary_key=???, time_column=None, end_time_column=None)[source]#
property name: str#

The name of this table.

property source_name: str#

The source name of this table.

has_column(name)[source]#

Returns True if this table holds a column with name name; False otherwise.

Return type:

bool

column(name)[source]#

Returns the data column named with name name in this table.

Parameters:

name (str) – The name of the column.

Raises:

KeyError – If name is not present in this table.

Return type:

Column

property columns: list[Column]#

Returns a list of Column objects that represent the columns in this table.

add_columns(columns)[source]#

Adds a set of columns to this table.

Parameters:

columns (Sequence[Union[ColumnSpec, Mapping[str, Any], str]]) – The columns to add.

Raises:

KeyError – If any of the column names already exist in this table.

Return type:

None

add_column(column)[source]#

Adds a column to this table.

Parameters:

column (Union[ColumnSpec, Mapping[str, Any], str]) – The column to add.

Raises:

KeyError – If the column name already exists in this table.

Return type:

Column

remove_column(name)[source]#

Removes a column from this table.

Parameters:

name (str) – The name of the column.

Raises:

KeyError – If name is not present in this table.

Return type:

Self

has_primary_key()[source]#

Returns True` if this table has a primary key; False otherwise.

Return type:

bool

property primary_key: Column | None#

The primary key column of this table.

The getter returns the primary key column of this table, or None if no such primary key is present.

The setter sets a column as a primary key on this table, and raises a ValueError if the primary key has a non-ID compatible data type or if the column name does not match a column in the data frame.

has_time_column()[source]#

Returns True if this table has a time column; False otherwise.

Return type:

bool

property time_column: Column | None#

The time column of this table.

The getter returns the time column of this table, or None if no such time column is present.

The setter sets a column as a time column on this table, and raises a ValueError if the time column has a non-timestamp compatible data type or if the column name does not match a column in the data frame.

has_end_time_column()[source]#

Returns True if this table has an end time column; False otherwise.

Return type:

bool

property end_time_column: Column | None#

The end time column of this table.

The getter returns the end time column of this table, or None if no such end time column is present.

The setter sets a column as an end time column on this table, and raises a ValueError if the end time column has a non-timestamp compatible data type or if the column name does not match a column in the data frame.

property metadata: DataFrame#

Returns a pandas.DataFrame object containing metadata information about the columns in this table.

The returned dataframe has columns "Name", "Data Type", "Semantic Type", "Primary Key", "Time Column" and "End Time Column", which provide an aggregated view of the properties of the columns of this table.

Example

>>> 
>>> import kumoai.experimental.rfm as rfm
>>> table = rfm.LocalTable(df=..., name=...).infer_metadata()
>>> table.metadata
    Name        Data Type  Semantic Type  Primary Key  Time Column  End Time Column
0   CustomerID  float64    ID             True         False        False
print_metadata()[source]#

Prints the metadata() of this table.

Return type:

None

infer_primary_key(verbose=True)[source]#

Infers the primary key in this table.

Parameters:

verbose (bool) – Whether to print verbose output.

Return type:

Self

infer_time_column(verbose=True)[source]#

Infers the time column in this table.

Parameters:

verbose (bool) – Whether to print verbose output.

Return type:

Self

infer_metadata(verbose=True)[source]#

Infers metadata, i.e., primary keys and time columns, in this table.

Parameters:

verbose (bool) – Whether to print verbose output.

Return type:

Self

abstract property backend: DataBackend#

The data backend of this table.

Contents