Use terminate()

VIDA-NYU · Feb 15, 2024 · bf4fe35 · bf4fe35
1 parent c4a0584
commit bf4fe35
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/alpha_automl/automl_manager.py b/alpha_automl/automl_manager.py
@@ -83,7 +83,7 @@ def _search_pipelines(self, automl_hyperparams):
             if result == 'DONE':
                 search_process.terminate()
                 search_process.join(10)
-                scoring_pool.close()
+                scoring_pool.terminate()
                 scoring_pool.join()
                 logger.debug(f'Found {self.found_pipelines} pipelines')
                 logger.debug('Search done')
@@ -122,7 +122,7 @@ def _search_pipelines(self, automl_hyperparams):
                         yield {'pipeline': pipeline, 'message': 'SCORED'}
                 else:
                     tmp_scoring_results.append(scoring_result)
-            
+
             scoring_results = tmp_scoring_results
 
             if time.time() > search_start_time + self.time_bound: