[BUG]The number of rows generating data is incorrect

[T0217]

Describe the bug
The number of rows of the generated synthetic data does not match the number of rows specified in the input. This issue currently occurs in the following models: 'cramer', 'dragan', 'gan', 'wgan', 'wgangp'.

To Reproduce
Run the file https://github.com/ydataai/ydata-synthetic/blob/dev/examples/regular/models/adult_wgangp.py

Screenshots

Desktop (please complete the following information):

• OS: Windows11
• Python: 3.11.5
• ydata-synthetic: 1.4.0

[Debasmith]

Hi! I’m new to contributing to ydata-synthetic but this bug looks interesting.
Can I take this up and work on a fix? @T0217

[T0217]

Hi! I’m new to contributing to ydata-synthetic but this bug looks interesting. Can I take this up and work on a fix? @T0217

Of course! If you need anything, feel free to reach out anytime.