KEP-2170: Add unit and integration tests for model and dataset initializers

[andreyvelich]

What you would like to be added?

We should add the unit tests and integration tests for dataset and model initializers after this PR: #2303

Love this feature?

Give it a 👍 We prioritize the features with most 👍

[seanlaii]

Hi, could I take this one?

[tenzen-y]

Hi, could I take this one?

Feel free to take this issue with /assign comment.

[seanlaii]

/assign

[seanlaii]

Could we rename this ticket to Add unit and Integration tests ...? Thank you!

[seanlaii]

Hi @andreyvelich , should we rename this issue by replacing e2e with integration?