RealTabFormer 0.2.4 causes integration to fail

[R-Palazzo]

Resolve #523
CU-86b81pd3u

Early in fit() there is a call to _train_with_sensitivity that leads to a few inconsistencies:
In _train_with_sensitivity there are:

gen_df = self.sample(
    n_samples=gen_rounds * gen_total, device=device, **gen_kwargs
)

While gen_kwargs is None by default. This is the error of the issue.

Then if save_full_every_epoch is not 0 then we try to save the model at some epoch. Their save method expect an self.experiment_id to be set while the attribute is set at the end of fit() only, so it crashes. The default value for save_full_every_epoch is 5

I hope they will fix those inconsistencies in future releases so we can return to the previous implementation.

@amontanez24 This highlights a case we might consider in the future, distinguishing model parameters from fit()/sample() parameters when defining a wrapper/variant. But I think it's out of scope for this issue.

[sdv-team]

Task linked: CU-86b81pd3u SDGym - RealTabFormer crashing with version #523

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 76.71%. Comparing base (b8b7b13) to head (a26868c).
⚠️ Report is 1 commits behind head on main.

Additional details and impacted files

@@           Coverage Diff           @@
##             main     #524   +/-   ##
=======================================
  Coverage   76.71%   76.71%           
=======================================
  Files          30       30           
  Lines        2456     2456           
=======================================
  Hits         1884     1884           
  Misses        572      572           

Flag	Coverage Δ
integration	62.47% <100.00%> (ø)
unit	70.27% <100.00%> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[amontanez24]

Can we file an issue on their repo?

[R-Palazzo]

Can we file an issue on their repo?

@amontanez24 yes I filed it:
worldbank/REaLTabFormer#103