This line shows how to save and load the information about each column's categorical precision:
https://github.com/SaferData/genjaxmix/blob/0cece36b6b30d015f863063764168f05e1453c8d/src/genspn/io.py#L50
however, the var_metadata key points to a dict of all the column names and their own parameters. This value means that undefined behavior will occur if the model is representing a column whose name is categorical_precisions. we should lift this key up a level.
However, we should do this outside of crunch time so we don't break existing schema loading.
This line shows how to save and load the information about each column's categorical precision:
https://github.com/SaferData/genjaxmix/blob/0cece36b6b30d015f863063764168f05e1453c8d/src/genspn/io.py#L50
however, the var_metadata key points to a dict of all the column names and their own parameters. This value means that undefined behavior will occur if the model is representing a column whose name is categorical_precisions. we should lift this key up a level.
However, we should do this outside of crunch time so we don't break existing schema loading.