bluecast.preprocessing.schema_checks

Module for DataFrame schema checks.

Module Contents

Classes

SchemaDetector

Detect and check DataFrame schema.

class bluecast.preprocessing.schema_checks.SchemaDetector

Detect and check DataFrame schema.

fit(df: pandas.DataFrame)

Store the schema of the train dataset.

transform(df: pandas.DataFrame)

Check if the test dataset has the same schema as the train dataset.

Will raise an error if schema length does not match and will raise a warning indicating the missing or extra columns.