New release.

Smoren · Nov 4, 2024 · 45feb04 · 45feb04
1 parent d99a8ae
commit 45feb04
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 3 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "genetic-search",
-  "version": "0.2.3",
+  "version": "0.2.4",
   "description": "Multiprocessing genetic algorithm implementation library",
   "license": "MIT",
   "repository": {

diff --git a/src/strategies.ts b/src/strategies.ts
@@ -48,7 +48,9 @@ export abstract class BaseRunnerStrategy<
   protected async execTask(inputs: TTaskConfig[]): Promise<GenerationGradeMatrix> {
     const result: GenerationGradeMatrix = [];
     for (const input of inputs) {
-      result.push(await this.config.task(input));
+      const taskResult = await this.config.task(input);
+      this.config.onTaskResult?.(taskResult);
+      result.push(taskResult);
     }
     return result;
   }
@@ -67,7 +69,9 @@ export abstract class BaseMultiprocessingRunnerStrategy<
 > extends BaseRunnerStrategy<TGenome, TConfig, TTaskConfig> {
   protected async execTask(inputs: TTaskConfig[]): Promise<GenerationGradeMatrix> {
     const pool = new Pool(this.config.poolSize);
-    const result: GenerationGradeMatrix = await pool.map(inputs, this.config.task);
+    const result: GenerationGradeMatrix = await pool.map(inputs, this.config.task, {
+      onResult: (result: any) => this.config.onTaskResult?.(result as GenomeGradeRow),
+    });
     pool.close();
 
     return result;

diff --git a/src/types.ts b/src/types.ts
@@ -35,6 +35,7 @@ export type BaseMutationStrategyConfig = {
 
 export type RunnerStrategyConfig<TTaskConfig> = {
   task: GradeGenerationTask<TTaskConfig>;
+  onTaskResult?: (result: GenomeGradeRow) => void;
 }
 
 export type MultiprocessingRunnerStrategyConfig<TTaskConfig> = RunnerStrategyConfig<TTaskConfig> & {

diff --git a/tests/parabola/multiprocess.test.ts b/tests/parabola/multiprocess.test.ts
@@ -32,6 +32,7 @@ describe('Parabola Multiprocessing', () => {
       runner: new ParabolaMultiprocessingRunnerStrategy({
         poolSize: 4,
         task: (x: ParabolaTaskConfig) => Promise.resolve([-((x[1]+12)**2) - 3]),
+        onTaskResult: () => void 0,
       }),
       scoring: new ParabolaTransparentScoringStrategy(),
       mutation: new ParabolaMutationStrategy(),

diff --git a/tests/parabola/single.test.ts b/tests/parabola/single.test.ts
@@ -35,6 +35,7 @@ describe.each([
         populate: new ParabolaPopulateStrategy(),
         runner: new ParabolaSingleRunnerStrategy({
           task: async (data: ParabolaTaskConfig) => [-((data[1]+a)**2) + b],
+          onTaskResult: (result) => void 0,
         }),
         scoring: new ParabolaTransparentScoringStrategy(),
         mutation: new ParabolaMutationStrategy(),