Skip to content

GitLab

Unverified Commit cda02aff authored by J-shang's avatar J-shang Committed by GitHub
Browse files

fix installable package meta (#3125) 


Co-authored-by: default avatarJ-shang <ningshang128@163.com>
parent 9d0b6fa6
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
nni/tools/nnictl/constants.py
View file @ cda02aff
...@@ -64,21 +64,21 @@ TRIAL_MONITOR_TAIL = '---------------------------------------------------------- ...@@ -64,21 +64,21 @@ TRIAL_MONITOR_TAIL = '----------------------------------------------------------
INSTALLABLE_PACKAGE_META = { INSTALLABLE_PACKAGE_META = {
'SMAC': { 'SMAC': {
'type': 'tuner', 'type': 'tuner',
'class_name': 'nni.smac_tuner.smac_tuner.SMACTuner', 'class_name': 'nni.algorithms.hpo.smac_tuner.smac_tuner.SMACTuner',
'code_sub_dir': 'smac_tuner', 'code_sub_dir': 'smac_tuner',
'class_args_validator': 'nni.smac_tuner.smac_tuner.SMACClassArgsValidator' 'class_args_validator': 'nni.algorithms.hpo.smac_tuner.smac_tuner.SMACClassArgsValidator'
}, },
'BOHB': { 'BOHB': {
'type': 'advisor', 'type': 'advisor',
'class_name': 'nni.bohb_advisor.bohb_advisor.BOHB', 'class_name': 'nni.algorithms.hpo.bohb_advisor.bohb_advisor.BOHB',
'code_sub_dir': 'bohb_advisor', 'code_sub_dir': 'bohb_advisor',
'class_args_validator': 'nni.bohb_advisor.bohb_advisor.BOHBClassArgsValidator' 'class_args_validator': 'nni.algorithms.hpo.bohb_advisor.bohb_advisor.BOHBClassArgsValidator'
}, },
'PPOTuner': { 'PPOTuner': {
'type': 'tuner', 'type': 'tuner',
'class_name': 'nni.ppo_tuner.ppo_tuner.PPOTuner', 'class_name': 'nni.algorithms.hpo.ppo_tuner.ppo_tuner.PPOTuner',
'code_sub_dir': 'ppo_tuner', 'code_sub_dir': 'ppo_tuner',
'class_args_validator': 'nni.ppo_tuner.ppo_tuner.PPOClassArgsValidator' 'class_args_validator': 'nni.algorithms.hpo.ppo_tuner.ppo_tuner.PPOClassArgsValidator'
} }
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment