In an interview with Nintendo Everything, the game's director, Masahiro Yamamoto, confirmed that the game takes place in a Kindergarten class within the Mario universe.
Contributed by GamerBen144