@epfml/discojs 2.1.2-p20240515133413.0 → 2.1.2-p20240531085945.0

package/dist/aggregator/base.js

@@ -24,6 +24,7 @@ export class Base {
      * It defines the effective aggregation group, which is possibly a subset
      * of all active nodes, depending on the aggregation scheme.
      */
+    // communication round -> NodeID -> T
     contributions;
     /**
      * Emits the aggregation event whenever an aggregation step is performed.

package/dist/aggregator/mean.d.ts

@@ -1,23 +1,18 @@
-import type { Map } from 'immutable';
-import { Base as Aggregator } from './base.js';
-import type { Model, WeightsContainer, client } from '../index.js';
-/**
- * Mean aggregator whose aggregation step consists in computing the mean of the received weights.
- */
+import type { Map } from "immutable";
+import { Base as Aggregator } from "./base.js";
+import type { Model, WeightsContainer, client } from "../index.js";
+/** Mean aggregator whose aggregation step consists in computing the mean of the received weights. */
 export declare class MeanAggregator extends Aggregator<WeightsContainer> {
+    #private;
     /**
-     * The threshold t to fulfill to trigger an aggregation step. It can either be:
-     * - relative: 0 < t <= 1, thus requiring t * |nodes| contributions
-     * - absolute: t > 1, thus requiring t contributions
+     * @param threshold - how many contributions for trigger an aggregation step.
+     *   - relative: 0 < t <= 1, thus requiring t * |nodes| contributions
+     *   - absolute: t > 1, thus requiring t contributions
      */
-    readonly threshold: number;
     constructor(model?: Model, roundCutoff?: number, threshold?: number);
-    /**
-     * Checks whether the contributions buffer is full, according to the set threshold.
-     * @returns Whether the contributions buffer is full
-     */
+    /** Checks whether the contributions buffer is full. */
     isFull(): boolean;
-    add(nodeId: client.NodeID, contribution: WeightsContainer, round: number): boolean;
+    add(nodeId: client.NodeID, contribution: WeightsContainer, round: number, currentContributions?: number): boolean;
     aggregate(): void;
     makePayloads(weights: WeightsContainer): Map<client.NodeID, WeightsContainer>;
 }

package/dist/aggregator/mean.js

@@ -1,65 +1,51 @@
-import { AggregationStep, Base as Aggregator } from './base.js';
-import { aggregation } from '../index.js';
-/**
- * Mean aggregator whose aggregation step consists in computing the mean of the received weights.
- */
+import { AggregationStep, Base as Aggregator } from "./base.js";
+import { aggregation } from "../index.js";
+/** Mean aggregator whose aggregation step consists in computing the mean of the received weights. */
 export class MeanAggregator extends Aggregator {
+    #threshold;
     /**
-     * The threshold t to fulfill to trigger an aggregation step. It can either be:
-     * - relative: 0 < t <= 1, thus requiring t * |nodes| contributions
-     * - absolute: t > 1, thus requiring t contributions
+     * @param threshold - how many contributions for trigger an aggregation step.
+     *   - relative: 0 < t <= 1, thus requiring t * |nodes| contributions
+     *   - absolute: t > 1, thus requiring t contributions
      */
-    threshold;
+    // TODO no way to require a single contribution
     constructor(model, roundCutoff = 0, threshold = 1) {
+        if (threshold <= 0)
+            throw new Error("threshold must be striclty positive");
+        if (threshold > 1 && !Number.isInteger(threshold))
+            throw new Error("absolute thresholds must be integeral");
         super(model, roundCutoff, 1);
-        // Default threshold is 100% of node participation
-        if (threshold === undefined) {
-            this.threshold = 1;
-            // Threshold must be positive
-        }
-        else if (threshold <= 0) {
-            throw new Error('threshold must be positive');
-            // Thresholds greater than 1 are considered absolute instead of relative to the number of nodes
-        }
-        else if (threshold > 1 && Math.round(threshold) !== threshold) {
-            throw new Error('absolute thresholds must integers');
-        }
-        else {
-            this.threshold = threshold;
-        }
+        this.#threshold = threshold;
     }
-    /**
-     * Checks whether the contributions buffer is full, according to the set threshold.
-     * @returns Whether the contributions buffer is full
-     */
+    /** Checks whether the contributions buffer is full. */
     isFull() {
-        if (this.threshold <= 1) {
-            const contribs = this.contributions.get(this.communicationRound);
-            if (contribs === undefined) {
-                return false;
-            }
-            return contribs.size >= this.threshold * this.nodes.size;
-        }
-        return this.contributions.size >= this.threshold;
+        const actualThreshold = this.#threshold <= 1
+            ? this.#threshold * this.nodes.size
+            : this.#threshold;
+        return (this.contributions.get(0)?.size ?? 0) >= actualThreshold;
     }
-    add(nodeId, contribution, round) {
-        if (this.nodes.has(nodeId) && this.isWithinRoundCutoff(round)) {
-            this.log(this.contributions.hasIn([0, nodeId]) ? AggregationStep.UPDATE : AggregationStep.ADD, nodeId);
-            this.contributions = this.contributions.setIn([0, nodeId], contribution);
-            this.informant?.update();
-            if (this.isFull()) {
-                this.aggregate();
-            }
-            return true;
-        }
-        return false;
+    add(nodeId, contribution, round, currentContributions = 0) {
+        if (currentContributions !== 0)
+            throw new Error("only a single communication round");
+        if (!this.nodes.has(nodeId) || !this.isWithinRoundCutoff(round))
+            return false;
+        this.log(this.contributions.hasIn([0, nodeId])
+            ? AggregationStep.UPDATE
+            : AggregationStep.ADD, nodeId);
+        this.contributions = this.contributions.setIn([0, nodeId], contribution);
+        this.informant?.update();
+        if (this.isFull())
+            this.aggregate();
+        return true;
     }
     aggregate() {
+        const currentContributions = this.contributions.get(0);
+        if (currentContributions === undefined)
+            throw new Error("aggregating without any contribution");
         this.log(AggregationStep.AGGREGATE);
-        const result = aggregation.avg(this.contributions.get(0)?.values());
-        if (this.model !== undefined) {
+        const result = aggregation.avg(currentContributions.values());
+        if (this.model !== undefined)
             this.model.weights = result;
-        }
         this.emit(result);
     }
     makePayloads(weights) {

package/dist/aggregator/secure.d.ts

@@ -1,6 +1,6 @@
-import { Map, List } from 'immutable';
-import { Base as Aggregator } from './base.js';
-import type { Model, WeightsContainer, client } from '../index.js';
+import { Map, List } from "immutable";
+import { Base as Aggregator } from "./base.js";
+import type { Model, WeightsContainer, client } from "../index.js";
 /**
  * Aggregator implementing secure multi-party computation for decentralized learning.
  * An aggregation consists of two communication rounds:
@@ -12,12 +12,10 @@ export declare class SecureAggregator extends Aggregator<WeightsContainer> {
     private readonly maxShareValue;
     constructor(model?: Model, maxShareValue?: number);
     aggregate(): void;
-    add(nodeId: client.NodeID, contribution: WeightsContainer, round: number, communicationRound: number): boolean;
+    add(nodeId: client.NodeID, contribution: WeightsContainer, round: number, communicationRound?: number): boolean;
     isFull(): boolean;
     makePayloads(weights: WeightsContainer): Map<client.NodeID, WeightsContainer>;
-    /**
-     * Generate N additive shares that aggregate to the secret weights array, where N is the number of peers.
-     */
+    /** Generate N additive shares that aggregate to the secret weights array, where N is the number of peers. */
     generateAllShares(secret: WeightsContainer): List<WeightsContainer>;
     /**
      * Generates one share in the same shape as the secret that is populated with values randomly chosen from

package/dist/aggregator/secure.js

@@ -1,7 +1,7 @@
-import { Map, List, Range } from 'immutable';
-import * as tf from '@tensorflow/tfjs';
-import { AggregationStep, Base as Aggregator } from './base.js';
-import { aggregation } from '../index.js';
+import { Map, List, Range } from "immutable";
+import * as tf from "@tensorflow/tfjs";
+import { AggregationStep, Base as Aggregator } from "./base.js";
+import { aggregation } from "../index.js";
 /**
  * Aggregator implementing secure multi-party computation for decentralized learning.
  * An aggregation consists of two communication rounds:
@@ -17,60 +17,72 @@ export class SecureAggregator extends Aggregator {
     }
     aggregate() {
         this.log(AggregationStep.AGGREGATE);
-        if (this.communicationRound === 0) {
+        switch (this.communicationRound) {
             // Sum the received shares
-            const result = aggregation.sum(this.contributions.get(0)?.values());
-            this.emit(result);
-        }
-        else if (this.communicationRound === 1) {
+            case 0: {
+                const currentContributions = this.contributions.get(0);
+                if (currentContributions === undefined)
+                    throw new Error("aggregating without any contribution");
+                const result = aggregation.sum(currentContributions.values());
+                this.emit(result);
+                break;
+            }
             // Average the received partial sums
-            const result = aggregation.avg(this.contributions.get(1)?.values());
-            if (this.model !== undefined) {
-                this.model.weights = result;
+            case 1: {
+                const currentContributions = this.contributions.get(1);
+                if (currentContributions === undefined)
+                    throw new Error("aggregating without any contribution");
+                const result = aggregation.avg(currentContributions.values());
+                if (this.model !== undefined)
+                    this.model.weights = result;
+                this.emit(result);
+                break;
             }
-            this.emit(result);
-        }
-        else {
-            throw new Error('communication round is out of bounds');
+            default:
+                throw new Error("communication round is out of bounds");
         }
     }
     add(nodeId, contribution, round, communicationRound) {
-        if (this.nodes.has(nodeId) && this.isWithinRoundCutoff(round)) {
-            this.log(this.contributions.hasIn([communicationRound, nodeId]) ? AggregationStep.UPDATE : AggregationStep.ADD, nodeId);
-            this.contributions = this.contributions.setIn([communicationRound, nodeId], contribution);
-            this.informant?.update();
-            if (this.isFull()) {
-                this.aggregate();
-            }
-            return true;
+        switch (communicationRound) {
+            case 0:
+            case 1:
+                break;
+            default:
+                throw new Error("requires communication round to be 0 or 1");
         }
-        return false;
+        if (!this.nodes.has(nodeId) || !this.isWithinRoundCutoff(round))
+            return false;
+        this.log(this.contributions.hasIn([communicationRound, nodeId])
+            ? AggregationStep.UPDATE
+            : AggregationStep.ADD, nodeId);
+        this.contributions = this.contributions.setIn([communicationRound, nodeId], contribution);
+        this.informant?.update();
+        if (this.isFull())
+            this.aggregate();
+        return true;
     }
     isFull() {
-        const contribs = this.contributions.get(this.communicationRound);
-        if (contribs === undefined) {
-            return false;
-        }
-        return contribs.size === this.nodes.size;
+        return ((this.contributions.get(this.communicationRound)?.size ?? 0) ===
+            this.nodes.size);
     }
     makePayloads(weights) {
-        if (this.communicationRound === 0) {
-            const shares = this.generateAllShares(weights);
-            // Abitrarily assign our shares to the available nodes
-            return Map(List(this.nodes).zip(shares));
-        }
-        else {
+        switch (this.communicationRound) {
+            case 0: {
+                const shares = this.generateAllShares(weights);
+                // Abitrarily assign our shares to the available nodes
+                return Map(List(this.nodes).zip(shares));
+            }
             // Send our partial sum to every other nodes
-            return this.nodes.toMap().map(() => weights);
+            case 1:
+                return this.nodes.toMap().map(() => weights);
+            default:
+                throw new Error("communication round is out of bounds");
         }
     }
-    /**
-     * Generate N additive shares that aggregate to the secret weights array, where N is the number of peers.
-     */
+    /** Generate N additive shares that aggregate to the secret weights array, where N is the number of peers. */
     generateAllShares(secret) {
-        if (this.nodes.size === 0) {
-            throw new Error('too few participants to generate shares');
-        }
+        if (this.nodes.size === 0)
+            throw new Error("too few participants to generate shares");
         // Generate N-1 shares
         const shares = Range(0, this.nodes.size - 1)
             .map(() => this.generateRandomShare(secret))
@@ -86,6 +98,6 @@ export class SecureAggregator extends Aggregator {
         const MAX_SEED_BITS = 47;
         const random = crypto.getRandomValues(new BigInt64Array(1))[0];
         const seed = Number(BigInt.asUintN(MAX_SEED_BITS, random));
-        return secret.map((t) => tf.randomUniform(t.shape, -this.maxShareValue, this.maxShareValue, 'float32', seed));
+        return secret.map((t) => tf.randomUniform(t.shape, -this.maxShareValue, this.maxShareValue, "float32", seed));
     }
 }

package/dist/client/federated/messages.d.ts

@@ -1,7 +1,7 @@
 import { type client, type MetadataKey, type MetadataValue } from '../../index.js';
 import { type weights } from '../../serialization/index.js';
 import { type, type AssignNodeID, type ClientConnected } from '../messages.js';
-export type MessageFederated = ClientConnected | SendPayload | ReceiveServerPayload | RequestServerStatistics | ReceiveServerStatistics | ReceiveServerMetadata | AssignNodeID;
+export type MessageFederated = ClientConnected | SendPayload | ReceiveServerPayload | ReceiveServerMetadata | AssignNodeID;
 export interface SendPayload {
     type: type.SendPayload;
     payload: weights.Encoded;
@@ -12,13 +12,6 @@ export interface ReceiveServerPayload {
     payload: weights.Encoded;
     round: number;
 }
-export interface RequestServerStatistics {
-    type: type.RequestServerStatistics;
-}
-export interface ReceiveServerStatistics {
-    type: type.ReceiveServerStatistics;
-    statistics: Record<string, number>;
-}
 export interface ReceiveServerMetadata {
     type: type.ReceiveServerMetadata;
     nodeId: client.NodeID;

package/dist/client/federated/messages.js

@@ -5,20 +5,11 @@ export function isMessageFederated(raw) {
     }
     switch (raw.type) {
         case type.ClientConnected:
-            return true;
         case type.SendPayload:
-            return true;
         case type.ReceiveServerPayload:
-            return true;
-        case type.RequestServerStatistics:
-            return true;
-        case type.ReceiveServerStatistics:
-            return true;
         case type.ReceiveServerMetadata:
-            return true;
         case type.AssignNodeID:
             return true;
-        default:
-            return false;
     }
+    return false;
 }

package/dist/client/messages.d.ts

@@ -10,9 +10,7 @@ export declare enum type {
     Payload = 5,
     SendPayload = 6,
     ReceiveServerMetadata = 7,
-    ReceiveServerPayload = 8,
-    RequestServerStatistics = 9,
-    ReceiveServerStatistics = 10
+    ReceiveServerPayload = 8
 }
 export interface ClientConnected {
     type: type.ClientConnected;

package/dist/client/messages.js

@@ -11,8 +11,6 @@ export var type;
     type[type["SendPayload"] = 6] = "SendPayload";
     type[type["ReceiveServerMetadata"] = 7] = "ReceiveServerMetadata";
     type[type["ReceiveServerPayload"] = 8] = "ReceiveServerPayload";
-    type[type["RequestServerStatistics"] = 9] = "RequestServerStatistics";
-    type[type["ReceiveServerStatistics"] = 10] = "ReceiveServerStatistics";
 })(type || (type = {}));
 export function hasMessageType(raw) {
     if (typeof raw !== 'object' || raw === null) {

package/dist/dataset/dataset_builder.d.ts

@@ -17,15 +17,11 @@ export declare class DatasetBuilder<Source> {
     /**
      * The buffer of unlabelled file sources.
      */
-    private _sources;
+    private _unlabeledSources;
     /**
      * The buffer of labelled file sources.
      */
-    private labelledSources;
-    /**
-     * Whether a dataset was already produced.
-     */
-    private _built;
+    private _labeledSources;
     constructor(
     /**
      * The data loader used to load the data contained in the provided files.
@@ -48,13 +44,8 @@ export declare class DatasetBuilder<Source> {
      * @param label The file sources label
      */
     clearFiles(label?: string): void;
-    private resetBuiltState;
     private getLabels;
     build(config?: DataConfig): Promise<DataSplit>;
-    /**
-     * Whether the dataset builder has already been consumed to produce a dataset.
-     */
-    get built(): boolean;
     get size(): number;
     get sources(): Source[];
 }

package/dist/dataset/dataset_builder.js

@@ -9,16 +9,11 @@ export class DatasetBuilder {
     /**
      * The buffer of unlabelled file sources.
      */
-    _sources;
+    _unlabeledSources;
     /**
      * The buffer of labelled file sources.
      */
-    labelledSources;
-    /**
-     * Whether a dataset was already produced.
-     */
-    // TODO useless, responsibility on callers
-    _built;
+    _labeledSources;
     constructor(
     /**
      * The data loader used to load the data contained in the provided files.
@@ -30,9 +25,9 @@ export class DatasetBuilder {
     task) {
         this.dataLoader = dataLoader;
         this.task = task;
-        this._sources = [];
-        this.labelledSources = Map();
-        this._built = false;
+        this._unlabeledSources = [];
+        // Map from label to sources
+        this._labeledSources = Map();
     }
     /**
      * Adds the given file sources to the builder's buffer. Sources may be provided a label in the case
@@ -41,19 +36,16 @@ export class DatasetBuilder {
      * @param label The file sources label
      */
     addFiles(sources, label) {
-        if (this.built) {
-            this.resetBuiltState();
-        }
         if (label === undefined) {
-            this._sources = this._sources.concat(sources);
+            this._unlabeledSources = this._unlabeledSources.concat(sources);
         }
         else {
-            const currentSources = this.labelledSources.get(label);
+            const currentSources = this._labeledSources.get(label);
             if (currentSources === undefined) {
-                this.labelledSources = this.labelledSources.set(label, sources);
+                this._labeledSources = this._labeledSources.set(label, sources);
             }
             else {
-                this.labelledSources = this.labelledSources.set(label, currentSources.concat(sources));
+                this._labeledSources = this._labeledSources.set(label, currentSources.concat(sources));
             }
         }
     }
@@ -63,27 +55,19 @@ export class DatasetBuilder {
      * @param label The file sources label
      */
     clearFiles(label) {
-        if (this.built) {
-            this.resetBuiltState();
-        }
         if (label === undefined) {
-            this._sources = [];
+            this._unlabeledSources = [];
         }
         else {
-            this.labelledSources = this.labelledSources.delete(label);
+            this._labeledSources = this._labeledSources.delete(label);
         }
     }
-    // If files are added or removed, then this should be called since the latest
-    // version of the dataset_builder has not yet been built.
-    resetBuiltState() {
-        this._built = false;
-    }
     getLabels() {
         // We need to duplicate the labels as we need one for each source.
         // Say for label A we have sources [img1, img2, img3], then we
         // need labels [A, A, A].
         let labels = [];
-        this.labelledSources.forEach((sources, label) => {
+        this._labeledSources.forEach((sources, label) => {
             const sourcesLabels = Array.from({ length: sources.length }, (_) => label);
             labels = labels.concat(sourcesLabels);
         });
@@ -91,17 +75,17 @@ export class DatasetBuilder {
     }
     async build(config) {
         // Require that at least one source collection is non-empty, but not both
-        if ((this._sources.length > 0) === (this.labelledSources.size > 0)) {
-            throw new Error('Please provide dataset input files'); // This error message is parsed in DatasetInput.vue
+        if (this._unlabeledSources.length + this._labeledSources.size === 0) {
+            throw new Error('No input files connected'); // This error message is parsed in Trainer.vue
         }
         let dataTuple;
-        if (this._sources.length > 0) {
+        if (this._unlabeledSources.length > 0) {
             let defaultConfig = {};
             if (config?.inference === true) {
                 // Inferring model, no labels needed
                 defaultConfig = {
                     features: this.task.trainingInformation.inputColumns,
-                    shuffle: false
+                    shuffle: true
                 };
             }
             else {
@@ -109,34 +93,26 @@ export class DatasetBuilder {
                 defaultConfig = {
                     features: this.task.trainingInformation.inputColumns,
                     labels: this.task.trainingInformation.outputColumns,
-                    shuffle: false
+                    shuffle: true
                 };
             }
-            dataTuple = await this.dataLoader.loadAll(this._sources, { ...defaultConfig, ...config });
+            dataTuple = await this.dataLoader.loadAll(this._unlabeledSources, { ...defaultConfig, ...config });
         }
         else {
             // Labels are inferred from the file selection boxes
             const defaultConfig = {
                 labels: this.getLabels(),
-                shuffle: false
+                shuffle: true
             };
-            const sources = this.labelledSources.valueSeq().toArray().flat();
+            const sources = this._labeledSources.valueSeq().toArray().flat();
             dataTuple = await this.dataLoader.loadAll(sources, { ...defaultConfig, ...config });
         }
-        // TODO @s314cy: Support .csv labels for image datasets (supervised training or testing)
-        this._built = true;
         return dataTuple;
     }
-    /**
-     * Whether the dataset builder has already been consumed to produce a dataset.
-     */
-    get built() {
-        return this._built;
-    }
     get size() {
-        return Math.max(this._sources.length, this.labelledSources.size);
+        return Math.max(this._unlabeledSources.length, this._labeledSources.size);
     }
     get sources() {
-        return this._sources.length > 0 ? this._sources : this.labelledSources.valueSeq().toArray().flat();
+        return this._unlabeledSources.length > 0 ? this._unlabeledSources : this._labeledSources.valueSeq().toArray().flat();
     }
 }

package/dist/default_tasks/cifar10.d.ts

@@ -0,0 +1,2 @@
+import type { TaskProvider } from '../index.js';
+export declare const cifar10: TaskProvider;

package/dist/default_tasks/{cifar10/index.js → cifar10.js}

@@ -1,6 +1,6 @@
 import * as tf from '@tensorflow/tfjs';
-import { data, models } from '../../index.js';
-import baseModel from './model.js';
+import { data, models } from '../index.js';
+import baseModel from '../models/mobileNet_v1_025_224.js';
 export const cifar10 = {
     getTask() {
         return {

package/dist/default_tasks/index.d.ts

@@ -1,6 +1,7 @@
-export { cifar10 } from './cifar10/index.js';
+export { cifar10 } from './cifar10.js';
 export { lusCovid } from './lus_covid.js';
+export { skinCondition } from './skin_condition.js';
 export { mnist } from './mnist.js';
-export { simpleFace } from './simple_face/index.js';
+export { simpleFace } from './simple_face.js';
 export { titanic } from './titanic.js';
 export { wikitext } from './wikitext.js';

package/dist/default_tasks/index.js

@@ -1,6 +1,7 @@
-export { cifar10 } from './cifar10/index.js';
+export { cifar10 } from './cifar10.js';
 export { lusCovid } from './lus_covid.js';
+export { skinCondition } from './skin_condition.js';
 export { mnist } from './mnist.js';
-export { simpleFace } from './simple_face/index.js';
+export { simpleFace } from './simple_face.js';
 export { titanic } from './titanic.js';
 export { wikitext } from './wikitext.js';

package/dist/default_tasks/lus_covid.js

@@ -8,7 +8,7 @@ export const lusCovid = {
                 taskTitle: 'COVID Lung Ultrasound',
                 summary: {
                     preview: 'Do you have a data of lung ultrasound images on patients <b>suspected of Lower Respiratory Tract infection (LRTI) during the COVID pandemic</b>? <br> Learn how to discriminate between COVID positive and negative patients by joining this task.',
-                    overview: "Don’t have a dataset of your own? Download a sample of a few cases <a class='underline' href='https://drive.switch.ch/index.php/s/zM5ZrUWK3taaIly' target='_blank'>here</a>."
+                    overview: "Don't have a dataset of your own? Download a sample of a few cases <a class='underline' href='https://drive.switch.ch/index.php/s/zM5ZrUWK3taaIly' target='_blank'>here</a>."
                 },
                 model: "We use a simplified* version of the <b>DeepChest model</b>: A deep learning model developed in our lab (<a class='underline' href='https://www.epfl.ch/labs/mlo/igh-intelligent-global-health/'>intelligent Global Health</a>.). On a cohort of 400 Swiss patients suspected of LRTI, the model obtained over 90% area under the ROC curve for this task. <br><br>*Simplified to ensure smooth running on your browser, the performance is minimally affected. Details of the adaptations are below <br>- <b>Removed</b>: positional embedding (i.e. we don’t take the anatomic position into consideration). Rather, the model now does mean pooling over the feature vector of the images for each patient <br>- <b>Replaced</b>: ResNet18 by Mobilenet",
                 dataFormatInformation: 'This model takes as input an image dataset. It consists on a set of lung ultrasound images per patient with its corresponding label of covid positive or negative. Moreover, to identify the images per patient you have to follow the follwing naming pattern: "patientId_*.png"',

package/dist/default_tasks/simple_face.d.ts

@@ -0,0 +1,2 @@
+import type { TaskProvider } from '../index.js';
+export declare const simpleFace: TaskProvider;

package/dist/default_tasks/{simple_face/index.js → simple_face.js}

@@ -1,6 +1,6 @@
 import * as tf from '@tensorflow/tfjs';
-import { data, models } from '../../index.js';
-import baseModel from './model.js';
+import { data, models } from '../index.js';
+import baseModel from '../models/mobileNetV2_35_alpha_2_classes.js';
 export const simpleFace = {
     getTask() {
         return {
@@ -12,7 +12,7 @@ export const simpleFace = {
                     overview: 'Simple face is a small subset of face_task from Kaggle'
                 },
                 dataFormatInformation: '',
-                dataExampleText: 'Below you find an example',
+                dataExampleText: 'Below you can find an example',
                 dataExampleImage: 'https://storage.googleapis.com/deai-313515.appspot.com/example_training_data/simple_face-example.png'
             },
             trainingInformation: {

package/dist/default_tasks/skin_condition.d.ts

@@ -0,0 +1,2 @@
+import type { TaskProvider } from '../index.js';
+export declare const skinCondition: TaskProvider;