Send number of cores

scripts/amlb/user_config/extensions/Alpha-AutoML/exec.py

@@ -38,17 +38,19 @@ def run(dataset, config):
     target_name = dataset.target.name
     output_path = config.output_dir
     time_bound = int(config.max_runtime_seconds/60)
+    cores = config.cores
 
     log.info(f'Received parameters:\n'
              f'train_dataset: {train_dataset_path}\n'
              f'test_dataset: {test_dataset_path}\n'
              f'target_name: {target_name}\n'
              f'time_bound: {time_bound}\n'
              f'metric: {metric}\n'
+             f'cores: {cores}'
              )
 
     automl = AutoMLClassifier(time_bound=time_bound, metric=metrics_mapping[metric], time_bound_run=15,
-                              output_folder=output_path, verbose=logging.DEBUG)
+                              output_folder=output_path, num_cpus=cores, verbose=logging.DEBUG)
 
     train_dataset = pd.read_csv(train_dataset_path)
     test_dataset = pd.read_csv(test_dataset_path)