[ML] Add integration tests for trained_models
API
#104819
+371
−35
Merged