def create_mock_experiment(): nnictl_experiment_config = Experiments() nnictl_experiment_config.add_experiment('xOpEwA5w', 8080, 123456, 'local', 'example_sklearn-classification') # mock process cmds = ['sleep', '3600000'] process = Popen(cmds, stdout=PIPE, stderr=STDOUT) nnictl_experiment_config.update_experiment('xOpEwA5w', 'pid', process.pid) nnictl_experiment_config.update_experiment('xOpEwA5w', 'port', 8080) nnictl_experiment_config.update_experiment('xOpEwA5w', 'webuiUrl', ['http://localhost:8080']) print_green("expriment start success, experiment id: xOpEwA5w")
def create_mock_experiment(): nnictl_experiment_config = Experiments() nnictl_experiment_config.add_experiment('xOpEwA5w', '8080', 123456, 'local', 'example_sklearn-classification') nni_config = Config('xOpEwA5w') # mock process cmds = ['sleep', '3600000'] process = Popen(cmds, stdout=PIPE, stderr=STDOUT) nni_config.set_config('restServerPid', process.pid) nni_config.set_config('experimentId', 'xOpEwA5w') nni_config.set_config('restServerPort', 8080) nni_config.set_config('webuiUrl', ['http://localhost:8080']) yml_path = Path(__file__).parents[1] / 'config_files/valid/test.yml' experiment_config = get_yml_content(str(yml_path)) nni_config.set_config('experimentConfig', experiment_config) print_green("expriment start success, experiment id: xOpEwA5w")
def test_update_experiment(self): experiment = Experiments(HOME_PATH) experiment.add_experiment('xOpEwA5w', 8081, 'N/A', 'local', 'test', endTime='N/A', status='INITIALIZED') self.assertTrue('xOpEwA5w' in experiment.get_all_experiments()) experiment.remove_experiment('xOpEwA5w') self.assertFalse('xOpEwA5w' in experiment.get_all_experiments())
def stop_mock_experiment(): config = Config('config') kill_command(config.get_config('restServerPid')) nnictl_experiment_config = Experiments() nnictl_experiment_config.remove_experiment('xOpEwA5w')
def stop_mock_experiment(): nnictl_experiment_config = Experiments() experiments_dict = nnictl_experiment_config.get_all_experiments() kill_command(experiments_dict['xOpEwA5w'].get('pid')) nnictl_experiment_config = Experiments() nnictl_experiment_config.remove_experiment('xOpEwA5w')