Trying to save a experiment_type:str to SQA store

[alxfed]

I'm trying to save an experiment_type string that has been assigned to the experiment_type property
of my Experiment to SQA Store.
It is declared as Optional[str] property in the signature of the Experiment class, and it assigns and reads beautifully. However, if you store the Experiment with this assigned (and present) experiment_type to the SQA store there is a 'null' in the column experiment_type of the experiment_v2 table. All the other fields are ORMapped to the table perfectly.

If you look at the properties of the class ax.storage.sqa_store.sqa_classes.SQAExperiment(*args: Any, **kwargs: Any)
experiment_type is Optional[int] = Column(None, Integer() there so it must be Enumed along the way somewhere

Should the experiment_types be declared or registered in some way before being used in Experiment classes so that they would store in the SQA store properly?

[lena-kashtelyan]

Hi, @alxfed! Great question. There are some docs on specifying experiment type here: https://ax.dev/docs/storage.html#customizing-1. In case a source code pointer is helpful, this is where the conversion to enum happens:

Ax/ax/storage/sqa_store/encoder.py

Lines 166 to 168 in 595095e

	experiment_type = self.get_enum_value(
	value=experiment.experiment_type, enum=self.config.experiment_type_enum
	)

Let me know if this doesn't help!

[alxfed]

Hi, again @lena-kashtelyan :) Thank you. That's exactly what I expected, but didn't manage to find this (really valuable for me) link. Thank you.

But I still can not make the save_experiment save this field to the SQA store table column.

P.S. Found it! the setting is in the SQAConfig. Got it. It would be good to mention it in the description of save_experiment somewhere... I think.

[lena-kashtelyan]

Good point!