diff --git a/discojs/src/dataset/data/preprocessing/text_preprocessing.spec.ts b/discojs/src/dataset/data/preprocessing/text_preprocessing.spec.ts index 4de9d8781..ea2f31e07 100644 --- a/discojs/src/dataset/data/preprocessing/text_preprocessing.spec.ts +++ b/discojs/src/dataset/data/preprocessing/text_preprocessing.spec.ts @@ -21,6 +21,7 @@ describe('text preprocessing', function () { validationSplit: 0, batchSize: 8, scheme: 'local', + minNbOfParticipants: 1, dataType: 'text', tokenizer: 'Xenova/gpt2', tensorBackend: 'gpt' diff --git a/docs/examples/custom_task.ts b/docs/examples/custom_task.ts index 08dfb502c..09e007c1b 100644 --- a/docs/examples/custom_task.ts +++ b/docs/examples/custom_task.ts @@ -30,6 +30,7 @@ const customTask: TaskProvider = { 'Output' ], scheme: 'federated', + minNbOfParticipants: 2, tensorBackend: 'tfjs', noiseScale: undefined, clippingRadius: undefined diff --git a/server/tests/client/federated.spec.ts b/server/tests/client/federated.spec.ts index aa23e0ed0..c0582b1b1 100644 --- a/server/tests/client/federated.spec.ts +++ b/server/tests/client/federated.spec.ts @@ -48,6 +48,7 @@ describe("federated client", () => { validationSplit: 0, batchSize: 1, scheme: "federated", + minNbOfParticipants: 2, dataType: "tabular", tensorBackend: 'tfjs' },