OmegaStore.put - wrong pipeline type
I am trying to invoke om.models.put with pipeline and model_store_name:
if model_type_name == "CTGAN": pipeline = CTGANSynthesizer() om.models.put(pipeline, model_store_name)
Which pass us to
omegaml.store.base.OmegaStore.putThis method put() takes as one of the args an
obj = <ctgan.synthesizer.CTGANSynthesizer at 0x7fd6423a2240>
So technically, this
obj is our
pipeline and a
CTGANSynthesizer instance at the same time.
But then, inside a method
OmegaStore.put we check:
- is_dataframe(obj) etc.
None of the checks perform
CTGANSynthesizer. As the result I get an error:
TypeError: type <class 'ctgan.synthesizer.CTGANSynthesizer'> not supported
Seems like there should be not just a
CTGANSynthesizer obj but some other type.
What type of pipeline should it bee?