Use lambda instead of functools.partial to create single-host config maker

[yhtang]

Fixes #1151

An alternative solution is to use functools.wraps but that requires a custom config validator:

axlearn.common.config.InvalidConfigValueError: Invalid config value type <class 'functools.partial'> for value "functools.partial(<function trainer_configs.<locals>.make_single_host_config at 0x722900936160>, 'fuji-3B-v3-flash')". Consider registering a custom validator with `register_validator`.

[Ethanlm]

I have not encountered this issue. How can I reproduce it?

[yhtang]

I have not encountered this issue. How can I reproduce it?

This is only triggered if using one of the *-single-host configs, e.g. fuji-3B-v3-flash-single-host

[Ethanlm]

I have not encountered this issue. How can I reproduce it?

This is only triggered if using one of the *-single-host configs, e.g. fuji-3B-v3-flash-single-host

Yeah I have been using single-host config for my testing on TPU recently.

[github-actions[bot]]

This pull request has been automatically marked as stale because it has been inactive for 60 days. It will be closed in 7 days if no further activity occurs. If you would like to continue working on this, please remove the stale label or leave a comment.