@epfml/discojs 3.0.1-p20250211140448.0 → 3.0.1-p20250218072819.0

package/dist/client/decentralized/peer.js

@@ -84,7 +84,8 @@ export class Peer {
         firstChunk.writeUint8(totalChunkCount, 3);
         b.copy(firstChunk, FIRST_HEADER_SIZE, 0, this.maxChunkSize - FIRST_HEADER_SIZE);
         return Seq.Indexed([firstChunk])
-            .concat(Range(1).zip(tail)
+            .concat(Range(1, Number.POSITIVE_INFINITY)
+            .zip(tail)
             .map(([id, raw]) => {
             const chunk = Buffer.alloc(HEADER_SIZE + raw.length);
             chunk.writeUint16BE(messageID);

package/dist/client/federated/federated_client.js

@@ -103,6 +103,8 @@ export class FederatedClient extends Client {
      * @returns the new global weights sent by the server
      */
     async onRoundEndCommunication(weights) {
+        if (this._ownId === undefined)
+            throw new Error("no received ID from server");
         if (this.aggregationResult === undefined) {
             throw new Error("local aggregation result was not set");
         }
@@ -111,7 +113,11 @@ export class FederatedClient extends Client {
         this.saveAndEmit("updating model");
         // Send our local contribution to the server
         // and receive the server global update for this round as an answer to our contribution
-        const payloadToServer = this.aggregator.makePayloads(weights).first();
+        const payloadToServer = this.aggregator
+            .makePayloads(weights)
+            .get(SERVER_NODE_ID);
+        if (payloadToServer === undefined)
+            throw new Error("aggregator didn't make a payload for the server");
         const msg = {
             type: type.SendPayload,
             payload: await serialization.weights.encode(payloadToServer),

package/dist/models/tfjs.js

@@ -26,7 +26,7 @@ export class TFJS extends Model {
     }
     async *train(trainingDataset, validationDataset) {
         let batchesLogs = List();
-        for await (const [batch, batchNumber] of trainingDataset.zip(Range())) {
+        for await (const [batch, batchNumber] of trainingDataset.zip(Range(0, Number.POSITIVE_INFINITY))) {
             const batchLogs = {
                 batch: batchNumber,
                 ...(await this.#runBatch(batch)),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@epfml/discojs",
-  "version": "3.0.1-p20250211140448.0",
+  "version": "3.0.1-p20250218072819.0",
   "type": "module",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -22,10 +22,9 @@
     "@epfml/isomorphic-wrtc": "1",
     "@jimp/core": "1",
     "@jimp/plugin-resize": "1",
-    "@msgpack/msgpack": "^3.0.0-beta2",
+    "@msgpack/msgpack": "<3.0.0-beta5",
     "@tensorflow/tfjs": "4",
     "@xenova/transformers": "2",
-    "immutable": "4",
     "isomorphic-ws": "5",
     "simple-peer": "9",
     "tslib": "2",