seismometer.data.loader.prediction.assumed_types

seismometer.data.loader.prediction.assumed_types(config, dataframe)

Convert the loaded predictions dataframe to the expected types.

Scope is currently restricted to time and output columns as parquet is expected to include datatypes.

Parameters:

• config (ConfigProvider) – The loaded configuration object.

• dataframe (pd.DataFrame) – The loaded predictions dataframe.

Returns:

The predictions dataframe with adjusted types.

Return type:

pd.DataFrame