Problem reshaping the data

[filipelauar]

Hi, I'm having troubles runing the code because of the function gm.fit(data[:, id_].reshape([-1, 1])) in the fit method of the BGMTransformer class, line 310.

It seems that the reshape made by the function data[:, id_].reshape([-1, 1]) leads to an error '(slice(None, None, None), 2)' is an invalid key. I tried to reshape by another methods like using loc or iloc, which works from terminal but I'm geting the same error when I change the function in the scrip. I don't know what to do now.

I loaded my own dataset and tried to run but I can't because of the error. That's what I am doing:

from sdgym.synthesizers.tvae import TVAESynthesizer
tvae = TVAESynthesizer()
tvae.fit(my_data, my_discrete_columns, my_ordinal_columns)