@epfml/discojs 1.0.0 → 2.0.0

package/dist/dataset/data_loader/image_loader.js

@@ -1,139 +1,26 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ImageLoader = void 0;
+exports.WebImageLoader = void 0;
 var tslib_1 = require("tslib");
-var immutable_1 = require("immutable");
 var __1 = require("../..");
-var data_loader_1 = require("./data_loader");
-/**
- * TODO @s314cy:
- * Load labels and correctly match them with their respective images, with the following constraints:
- * 1. Images are given as 1 image/1 file
- * 2. Labels are given as multiple labels/1 file, each label file can contain a different amount of labels
- */
-var ImageLoader = /** @class */ (function (_super) {
-    (0, tslib_1.__extends)(ImageLoader, _super);
-    function ImageLoader() {
+var WebImageLoader = /** @class */ (function (_super) {
+    (0, tslib_1.__extends)(WebImageLoader, _super);
+    function WebImageLoader() {
         return _super !== null && _super.apply(this, arguments) || this;
     }
-    ImageLoader.prototype.load = function (image, config) {
+    WebImageLoader.prototype.readImageFrom = function (source) {
         return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var tensorContainer;
-            var _a;
-            return (0, tslib_1.__generator)(this, function (_b) {
-                switch (_b.label) {
-                    case 0:
-                        if (!(config === undefined || config.labels === undefined)) return [3 /*break*/, 2];
-                        return [4 /*yield*/, this.readImageFrom(image)];
-                    case 1:
-                        tensorContainer = _b.sent();
-                        return [3 /*break*/, 4];
-                    case 2:
-                        _a = {};
-                        return [4 /*yield*/, this.readImageFrom(image)];
-                    case 3:
-                        tensorContainer = (_a.xs = _b.sent(),
-                            _a.ys = config.labels[0],
-                            _a);
-                        _b.label = 4;
-                    case 4: return [2 /*return*/, __1.tf.data.array([tensorContainer])];
-                }
-            });
-        });
-    };
-    ImageLoader.prototype.buildDataset = function (images, labels, indices, config) {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var dataset;
-            var _this = this;
-            return (0, tslib_1.__generator)(this, function (_a) {
-                dataset = __1.tf.data.generator(function () {
-                    var withLabels = (config === null || config === void 0 ? void 0 : config.labels) !== undefined;
-                    var index = 0;
-                    var iterator = {
-                        next: function () { return (0, tslib_1.__awaiter)(_this, void 0, void 0, function () {
-                            var sample, label, value;
-                            return (0, tslib_1.__generator)(this, function (_a) {
-                                switch (_a.label) {
-                                    case 0:
-                                        if (index === indices.length) {
-                                            return [2 /*return*/, { done: true }];
-                                        }
-                                        return [4 /*yield*/, this.readImageFrom(images[indices[index]])];
-                                    case 1:
-                                        sample = _a.sent();
-                                        label = withLabels ? labels[indices[index]] : undefined;
-                                        value = withLabels ? { xs: sample, ys: label } : sample;
-                                        index++;
-                                        return [2 /*return*/, {
-                                                value: value,
-                                                done: false
-                                            }];
-                                }
-                            });
-                        }); }
-                    };
-                    return iterator; // Lazy
-                });
-                return [2 /*return*/, {
-                        dataset: dataset,
-                        size: indices.length
-                    }];
-            });
-        });
-    };
-    ImageLoader.prototype.loadAll = function (images, config) {
-        var _a, _b;
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var labels, indices, numberOfClasses, dataset, trainSize, trainIndices, valIndices, trainDataset, valDataset;
+            var _a, _b;
             return (0, tslib_1.__generator)(this, function (_c) {
                 switch (_c.label) {
                     case 0:
-                        labels = [];
-                        indices = (0, immutable_1.Range)(0, images.length).toArray();
-                        if ((config === null || config === void 0 ? void 0 : config.labels) !== undefined) {
-                            numberOfClasses = (_b = (_a = this.task.trainingInformation) === null || _a === void 0 ? void 0 : _a.LABEL_LIST) === null || _b === void 0 ? void 0 : _b.length;
-                            if (numberOfClasses === undefined) {
-                                throw new Error('wanted labels but none found in task');
-                            }
-                            labels = __1.tf.oneHot(__1.tf.tensor1d(config.labels, 'int32'), numberOfClasses).arraySync();
-                        }
-                        if ((config === null || config === void 0 ? void 0 : config.shuffle) === undefined || (config === null || config === void 0 ? void 0 : config.shuffle)) {
-                            this.shuffle(indices);
-                        }
-                        if (!((config === null || config === void 0 ? void 0 : config.validationSplit) === undefined)) return [3 /*break*/, 2];
-                        return [4 /*yield*/, this.buildDataset(images, labels, indices, config)];
-                    case 1:
-                        dataset = _c.sent();
-                        return [2 /*return*/, {
-                                train: dataset,
-                                validation: dataset
-                            }];
-                    case 2:
-                        trainSize = Math.floor(images.length * (1 - config.validationSplit));
-                        trainIndices = indices.slice(0, trainSize);
-                        valIndices = indices.slice(trainSize);
-                        return [4 /*yield*/, this.buildDataset(images, labels, trainIndices, config)];
-                    case 3:
-                        trainDataset = _c.sent();
-                        return [4 /*yield*/, this.buildDataset(images, labels, valIndices, config)];
-                    case 4:
-                        valDataset = _c.sent();
-                        return [2 /*return*/, {
-                                train: trainDataset,
-                                validation: valDataset
-                            }];
+                        _b = (_a = __1.tf.browser).fromPixels;
+                        return [4 /*yield*/, createImageBitmap(source)];
+                    case 1: return [2 /*return*/, _b.apply(_a, [_c.sent()])];
                 }
             });
         });
     };
-    ImageLoader.prototype.shuffle = function (array) {
-        for (var i = 0; i < array.length; i++) {
-            var j = Math.floor(Math.random() * i);
-            var swap = array[i];
-            array[i] = array[j];
-            array[j] = swap;
-        }
-    };
-    return ImageLoader;
-}(data_loader_1.DataLoader));
-exports.ImageLoader = ImageLoader;
+    return WebImageLoader;
+}(__1.data.ImageLoader));
+exports.WebImageLoader = WebImageLoader;

package/dist/dataset/data_loader/index.d.ts

@@ -1,3 +1,2 @@
-export { Data, DataTuple, DataConfig, DataLoader } from './data_loader';
-export { ImageLoader } from './image_loader';
-export { TabularLoader } from './tabular_loader';
+export { WebImageLoader } from './image_loader';
+export { WebTabularLoader } from './tabular_loader';

package/dist/dataset/data_loader/index.js

@@ -1,9 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.TabularLoader = exports.ImageLoader = exports.DataLoader = void 0;
-var data_loader_1 = require("./data_loader");
-Object.defineProperty(exports, "DataLoader", { enumerable: true, get: function () { return data_loader_1.DataLoader; } });
+exports.WebTabularLoader = exports.WebImageLoader = void 0;
 var image_loader_1 = require("./image_loader");
-Object.defineProperty(exports, "ImageLoader", { enumerable: true, get: function () { return image_loader_1.ImageLoader; } });
+Object.defineProperty(exports, "WebImageLoader", { enumerable: true, get: function () { return image_loader_1.WebImageLoader; } });
 var tabular_loader_1 = require("./tabular_loader");
-Object.defineProperty(exports, "TabularLoader", { enumerable: true, get: function () { return tabular_loader_1.TabularLoader; } });
+Object.defineProperty(exports, "WebTabularLoader", { enumerable: true, get: function () { return tabular_loader_1.WebTabularLoader; } });

package/dist/dataset/data_loader/tabular_loader.d.ts

@@ -1,29 +1,4 @@
-import { DataLoader, DataConfig, DataTuple } from './data_loader';
-import { Dataset } from '../dataset_builder';
-import { Task } from '../../task';
-import { tf } from '../..';
-export declare abstract class TabularLoader<Source> extends DataLoader<Source> {
-    private readonly delimiter;
-    constructor(task: Task, delimiter: string);
-    /**
-     * Creates a CSV dataset object based off the given source.
-     * @param source File object, URL string or local file system path.
-     * @param csvConfig Object expected by TF.js to create a CSVDataset.
-     * @returns The CSVDataset object built upon the given source.
-     */
-    abstract loadTabularDatasetFrom(source: Source, csvConfig: Record<string, unknown>): tf.data.CSVDataset;
-    /**
-     * Expects delimiter-separated tabular data made of N columns. The data may be
-     * potentially split among several sources. Every source should contain N-1
-     * feature columns and 1 single label column.
-     * @param source List of File objects, URLs or file system paths.
-     * @param config
-     * @returns A TF.js dataset built upon read tabular data stored in the given sources.
-     */
-    load(source: Source, config?: DataConfig): Promise<Dataset>;
-    /**
-      * Creates the CSV datasets based off the given sources, then fuses them into a single CSV
-      * dataset.
-      */
-    loadAll(sources: Source[], config: DataConfig): Promise<DataTuple>;
+import { tf, data } from '../..';
+export declare class WebTabularLoader extends data.TabularLoader<File> {
+    loadTabularDatasetFrom(source: File, csvConfig: Record<string, unknown>): tf.data.CSVDataset;
 }

package/dist/dataset/data_loader/tabular_loader.js

@@ -1,97 +1,16 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.TabularLoader = void 0;
+exports.WebTabularLoader = void 0;
 var tslib_1 = require("tslib");
-var data_loader_1 = require("./data_loader");
-var immutable_1 = require("immutable");
-// window size from which the dataset shuffling will sample
-var BUFFER_SIZE = 1000;
-var TabularLoader = /** @class */ (function (_super) {
-    (0, tslib_1.__extends)(TabularLoader, _super);
-    function TabularLoader(task, delimiter) {
-        var _this = _super.call(this, task) || this;
-        _this.delimiter = delimiter;
-        return _this;
+var __1 = require("../..");
+var WebTabularLoader = /** @class */ (function (_super) {
+    (0, tslib_1.__extends)(WebTabularLoader, _super);
+    function WebTabularLoader() {
+        return _super !== null && _super.apply(this, arguments) || this;
     }
-    /**
-     * Expects delimiter-separated tabular data made of N columns. The data may be
-     * potentially split among several sources. Every source should contain N-1
-     * feature columns and 1 single label column.
-     * @param source List of File objects, URLs or file system paths.
-     * @param config
-     * @returns A TF.js dataset built upon read tabular data stored in the given sources.
-     */
-    TabularLoader.prototype.load = function (source, config) {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var columnConfigs, csvConfig, dataset;
-            return (0, tslib_1.__generator)(this, function (_a) {
-                /**
-                 * Prepare the CSV config object based off the given features and labels.
-                 * If labels is empty, then the returned dataset is comprised of samples only.
-                 * Otherwise, each entry is of the form `{ xs, ys }` with `xs` as features and `ys`
-                 * as labels.
-                 */
-                if ((config === null || config === void 0 ? void 0 : config.features) === undefined) {
-                    // TODO @s314cy
-                    throw new Error('not implemented');
-                }
-                columnConfigs = (0, immutable_1.Map)((0, immutable_1.Set)(config.features).map(function (feature) { return [feature, { required: false, isLabel: false }]; })).merge((0, immutable_1.Set)(config.labels).map(function (label) { return [label, { required: true, isLabel: true }]; }));
-                csvConfig = {
-                    hasHeader: true,
-                    columnConfigs: columnConfigs.toObject(),
-                    configuredColumnsOnly: true,
-                    delimiter: this.delimiter
-                };
-                dataset = this.loadTabularDatasetFrom(source, csvConfig).map(function (t) {
-                    if (typeof t === 'object' && ('xs' in t) && ('ys' in t)) {
-                        return t;
-                    }
-                    throw new Error('expected TensorContainerObject');
-                }).map(function (t) {
-                    // TODO order may not be stable between tensor
-                    var _a = t, xs = _a.xs, ys = _a.ys;
-                    return {
-                        xs: Object.values(xs),
-                        ys: Object.values(ys)
-                    };
-                });
-                return [2 /*return*/, ((config === null || config === void 0 ? void 0 : config.shuffle) === undefined || (config === null || config === void 0 ? void 0 : config.shuffle)) ? dataset.shuffle(BUFFER_SIZE) : dataset];
-            });
-        });
+    WebTabularLoader.prototype.loadTabularDatasetFrom = function (source, csvConfig) {
+        return new __1.tf.data.CSVDataset(new __1.tf.data.FileDataSource(source), csvConfig);
     };
-    /**
-      * Creates the CSV datasets based off the given sources, then fuses them into a single CSV
-      * dataset.
-      */
-    TabularLoader.prototype.loadAll = function (sources, config) {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var datasets, dataset, data;
-            var _this = this;
-            return (0, tslib_1.__generator)(this, function (_a) {
-                switch (_a.label) {
-                    case 0: return [4 /*yield*/, Promise.all(sources.map(function (source) { return (0, tslib_1.__awaiter)(_this, void 0, void 0, function () { return (0, tslib_1.__generator)(this, function (_a) {
-                            switch (_a.label) {
-                                case 0: return [4 /*yield*/, this.load(source, (0, tslib_1.__assign)((0, tslib_1.__assign)({}, config), { shuffle: false }))];
-                                case 1: return [2 /*return*/, _a.sent()];
-                            }
-                        }); }); }))];
-                    case 1:
-                        datasets = _a.sent();
-                        dataset = (0, immutable_1.List)(datasets).reduce(function (acc, dataset) { return acc.concatenate(dataset); });
-                        data = {
-                            dataset: (config === null || config === void 0 ? void 0 : config.shuffle) ? dataset.shuffle(BUFFER_SIZE) : dataset,
-                            // dataset.size does not work for csv datasets
-                            // https://github.com/tensorflow/tfjs/issues/5845
-                            size: 0
-                        };
-                        // TODO: Implement validation split for tabular data (tricky due to streaming)
-                        return [2 /*return*/, {
-                                train: data
-                            }];
-                }
-            });
-        });
-    };
-    return TabularLoader;
-}(data_loader_1.DataLoader));
-exports.TabularLoader = TabularLoader;
+    return WebTabularLoader;
+}(__1.data.TabularLoader));
+exports.WebTabularLoader = WebTabularLoader;

package/dist/imports.d.ts

@@ -0,0 +1,2 @@
+export * as data from './dataset/data_loader';
+export { IndexedDB } from './memory';

package/dist/imports.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IndexedDB = exports.data = void 0;
+var tslib_1 = require("tslib");
+exports.data = (0, tslib_1.__importStar)(require("./dataset/data_loader"));
+var memory_1 = require("./memory");
+Object.defineProperty(exports, "IndexedDB", { enumerable: true, get: function () { return memory_1.IndexedDB; } });

package/dist/index.d.ts

@@ -1,19 +1,2 @@
-export * as aggregation from './aggregation';
-export * as dataset from './dataset';
-export * as serialization from './serialization';
-export * as tasks from './tasks';
-export * as training from './training';
-export * as privacy from './privacy';
-export { GraphInformant, TrainingInformant, informant } from './informant';
-export { Base as Client } from './client';
-export * as client from './client';
-export { AsyncBuffer } from './async_buffer';
-export { AsyncInformant } from './async_informant';
-export { Logger, ConsoleLogger, TrainerLog } from './logging';
-export { Memory, ModelType, ModelInfo, Path, ModelSource, Empty as EmptyMemory } from './memory';
-export { ModelActor } from './model_actor';
-export { Disco, TrainingSchemes } from './training';
-export { Validator } from './validation';
-export { TrainingInformation, DisplayInformation, isTask, Task, isTaskID, TaskID } from './task';
-export * from './types';
-export { tf } from './tfjs';
+export * from './core';
+export * as browser from './imports';

package/dist/index.js

@@ -1,42 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.tf = exports.isTaskID = exports.isTask = exports.Validator = exports.TrainingSchemes = exports.Disco = exports.ModelActor = exports.EmptyMemory = exports.ModelType = exports.Memory = exports.TrainerLog = exports.ConsoleLogger = exports.Logger = exports.AsyncInformant = exports.AsyncBuffer = exports.client = exports.Client = exports.informant = exports.TrainingInformant = exports.GraphInformant = exports.privacy = exports.training = exports.tasks = exports.serialization = exports.dataset = exports.aggregation = void 0;
+exports.browser = void 0;
 var tslib_1 = require("tslib");
-exports.aggregation = (0, tslib_1.__importStar)(require("./aggregation"));
-exports.dataset = (0, tslib_1.__importStar)(require("./dataset"));
-exports.serialization = (0, tslib_1.__importStar)(require("./serialization"));
-exports.tasks = (0, tslib_1.__importStar)(require("./tasks"));
-exports.training = (0, tslib_1.__importStar)(require("./training"));
-exports.privacy = (0, tslib_1.__importStar)(require("./privacy"));
-var informant_1 = require("./informant");
-Object.defineProperty(exports, "GraphInformant", { enumerable: true, get: function () { return informant_1.GraphInformant; } });
-Object.defineProperty(exports, "TrainingInformant", { enumerable: true, get: function () { return informant_1.TrainingInformant; } });
-Object.defineProperty(exports, "informant", { enumerable: true, get: function () { return informant_1.informant; } });
-var client_1 = require("./client");
-Object.defineProperty(exports, "Client", { enumerable: true, get: function () { return client_1.Base; } });
-exports.client = (0, tslib_1.__importStar)(require("./client"));
-var async_buffer_1 = require("./async_buffer");
-Object.defineProperty(exports, "AsyncBuffer", { enumerable: true, get: function () { return async_buffer_1.AsyncBuffer; } });
-var async_informant_1 = require("./async_informant");
-Object.defineProperty(exports, "AsyncInformant", { enumerable: true, get: function () { return async_informant_1.AsyncInformant; } });
-var logging_1 = require("./logging");
-Object.defineProperty(exports, "Logger", { enumerable: true, get: function () { return logging_1.Logger; } });
-Object.defineProperty(exports, "ConsoleLogger", { enumerable: true, get: function () { return logging_1.ConsoleLogger; } });
-Object.defineProperty(exports, "TrainerLog", { enumerable: true, get: function () { return logging_1.TrainerLog; } });
-var memory_1 = require("./memory");
-Object.defineProperty(exports, "Memory", { enumerable: true, get: function () { return memory_1.Memory; } });
-Object.defineProperty(exports, "ModelType", { enumerable: true, get: function () { return memory_1.ModelType; } });
-Object.defineProperty(exports, "EmptyMemory", { enumerable: true, get: function () { return memory_1.Empty; } });
-var model_actor_1 = require("./model_actor");
-Object.defineProperty(exports, "ModelActor", { enumerable: true, get: function () { return model_actor_1.ModelActor; } });
-var training_1 = require("./training");
-Object.defineProperty(exports, "Disco", { enumerable: true, get: function () { return training_1.Disco; } });
-Object.defineProperty(exports, "TrainingSchemes", { enumerable: true, get: function () { return training_1.TrainingSchemes; } });
-var validation_1 = require("./validation");
-Object.defineProperty(exports, "Validator", { enumerable: true, get: function () { return validation_1.Validator; } });
-var task_1 = require("./task");
-Object.defineProperty(exports, "isTask", { enumerable: true, get: function () { return task_1.isTask; } });
-Object.defineProperty(exports, "isTaskID", { enumerable: true, get: function () { return task_1.isTaskID; } });
-(0, tslib_1.__exportStar)(require("./types"), exports);
-var tfjs_1 = require("./tfjs");
-Object.defineProperty(exports, "tf", { enumerable: true, get: function () { return tfjs_1.tf; } });
+(0, tslib_1.__exportStar)(require("./core"), exports);
+exports.browser = (0, tslib_1.__importStar)(require("./imports"));

package/dist/memory/index.d.ts

@@ -1,3 +1 @@
-export { Empty } from './empty';
-export { Memory, ModelInfo, Path, ModelSource } from './base';
-export { ModelType } from './model_type';
+export { IndexedDB } from './memory';

package/dist/memory/index.js

@@ -1,9 +1,5 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ModelType = exports.Memory = exports.Empty = void 0;
-var empty_1 = require("./empty");
-Object.defineProperty(exports, "Empty", { enumerable: true, get: function () { return empty_1.Empty; } });
-var base_1 = require("./base");
-Object.defineProperty(exports, "Memory", { enumerable: true, get: function () { return base_1.Memory; } });
-var model_type_1 = require("./model_type");
-Object.defineProperty(exports, "ModelType", { enumerable: true, get: function () { return model_type_1.ModelType; } });
+exports.IndexedDB = void 0;
+var memory_1 = require("./memory");
+Object.defineProperty(exports, "IndexedDB", { enumerable: true, get: function () { return memory_1.IndexedDB; } });

package/dist/memory/memory.d.ts

@@ -0,0 +1,26 @@
+import { tf, Memory, Path, ModelInfo, ModelSource } from '..';
+export declare class IndexedDB extends Memory {
+    pathFor(source: ModelSource): Path;
+    infoFor(source: ModelSource): ModelInfo;
+    getModelMetadata(source: ModelSource): Promise<tf.io.ModelArtifactsInfo | undefined>;
+    contains(source: ModelSource): Promise<boolean>;
+    getModel(source: ModelSource): Promise<tf.LayersModel>;
+    deleteModel(source: ModelSource): Promise<void>;
+    loadModel(source: ModelSource): Promise<void>;
+    /**
+     * Saves the working model to the source.
+     * @param source the destination
+     * @param model the model
+     */
+    updateWorkingModel(source: ModelSource, model: tf.LayersModel): Promise<void>;
+    /**
+   * Creates a saved copy of the working model corresponding to the source.
+   * @param source the source
+   */
+    saveWorkingModel(source: ModelSource): Promise<void>;
+    /**
+   * Downloads the model corresponding to the source.
+   * @param source the source
+   */
+    downloadModel(source: ModelSource): Promise<void>;
+}

package/dist/memory/memory.js

@@ -0,0 +1,160 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IndexedDB = void 0;
+var tslib_1 = require("tslib");
+/**
+ * Helper functions used to load and save TFJS models from IndexedDB. The
+ * working model is the model currently being trained for a task. Saved models
+ * are models that were explicitly saved to IndexedDB. The two working/ and saved/
+ * folders are invisible to the user. The user only interacts with the saved/
+ * folder via the model library. The working/ folder is only used by the backend.
+ * The working model is loaded from IndexedDB for training (model.fit) only.
+ */
+var path_1 = (0, tslib_1.__importDefault)(require("path"));
+var __1 = require("..");
+var IndexedDB = /** @class */ (function (_super) {
+    (0, tslib_1.__extends)(IndexedDB, _super);
+    function IndexedDB() {
+        return _super !== null && _super.apply(this, arguments) || this;
+    }
+    IndexedDB.prototype.pathFor = function (source) {
+        if (typeof source === 'string') {
+            return source;
+        }
+        if (source.type === undefined || source.taskID === undefined || source.name === undefined) {
+            throw new TypeError('source incomplete');
+        }
+        return 'indexeddb://' + path_1.default.join(source.type, source.taskID, source.name);
+    };
+    IndexedDB.prototype.infoFor = function (source) {
+        if (typeof source !== 'string') {
+            return source;
+        }
+        var _a = (0, tslib_1.__read)(source.split('/').splice(2), 3), stringType = _a[0], taskID = _a[1], name = _a[2];
+        var type = stringType === 'working' ? __1.ModelType.WORKING : __1.ModelType.SAVED;
+        return { type: type, taskID: taskID, name: name };
+    };
+    IndexedDB.prototype.getModelMetadata = function (source) {
+        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
+            var models;
+            return (0, tslib_1.__generator)(this, function (_a) {
+                switch (_a.label) {
+                    case 0: return [4 /*yield*/, __1.tf.io.listModels()];
+                    case 1:
+                        models = _a.sent();
+                        return [2 /*return*/, models[this.pathFor(source)]];
+                }
+            });
+        });
+    };
+    IndexedDB.prototype.contains = function (source) {
+        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
+            return (0, tslib_1.__generator)(this, function (_a) {
+                switch (_a.label) {
+                    case 0: return [4 /*yield*/, this.getModelMetadata(source)];
+                    case 1: return [2 /*return*/, (_a.sent()) !== undefined];
+                }
+            });
+        });
+    };
+    IndexedDB.prototype.getModel = function (source) {
+        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
+            return (0, tslib_1.__generator)(this, function (_a) {
+                switch (_a.label) {
+                    case 0: return [4 /*yield*/, __1.tf.loadLayersModel(this.pathFor(source))];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        });
+    };
+    IndexedDB.prototype.deleteModel = function (source) {
+        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
+            return (0, tslib_1.__generator)(this, function (_a) {
+                switch (_a.label) {
+                    case 0: return [4 /*yield*/, __1.tf.io.removeModel(this.pathFor(source))];
+                    case 1:
+                        _a.sent();
+                        return [2 /*return*/];
+                }
+            });
+        });
+    };
+    IndexedDB.prototype.loadModel = function (source) {
+        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
+            return (0, tslib_1.__generator)(this, function (_a) {
+                switch (_a.label) {
+                    case 0: return [4 /*yield*/, __1.tf.io.copyModel(this.pathFor(source), this.pathFor((0, tslib_1.__assign)((0, tslib_1.__assign)({}, this.infoFor(source)), { type: __1.ModelType.WORKING })))];
+                    case 1:
+                        _a.sent();
+                        return [2 /*return*/];
+                }
+            });
+        });
+    };
+    /**
+     * Saves the working model to the source.
+     * @param source the destination
+     * @param model the model
+     */
+    IndexedDB.prototype.updateWorkingModel = function (source, model) {
+        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
+            var src;
+            return (0, tslib_1.__generator)(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        src = this.infoFor(source);
+                        if (src.type !== undefined && src.type !== __1.ModelType.WORKING) {
+                            throw new TypeError('expected working model');
+                        }
+                        return [4 /*yield*/, model.save(this.pathFor((0, tslib_1.__assign)((0, tslib_1.__assign)({}, src), { type: __1.ModelType.WORKING })))];
+                    case 1:
+                        _a.sent();
+                        return [2 /*return*/];
+                }
+            });
+        });
+    };
+    /**
+   * Creates a saved copy of the working model corresponding to the source.
+   * @param source the source
+   */
+    IndexedDB.prototype.saveWorkingModel = function (source) {
+        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
+            var src;
+            return (0, tslib_1.__generator)(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        src = this.infoFor(source);
+                        if (src.type !== undefined && src.type !== __1.ModelType.WORKING) {
+                            throw new TypeError('expected working model');
+                        }
+                        return [4 /*yield*/, __1.tf.io.copyModel(this.pathFor((0, tslib_1.__assign)((0, tslib_1.__assign)({}, src), { type: __1.ModelType.WORKING })), this.pathFor((0, tslib_1.__assign)((0, tslib_1.__assign)({}, src), { type: __1.ModelType.SAVED })))];
+                    case 1:
+                        _a.sent();
+                        return [2 /*return*/];
+                }
+            });
+        });
+    };
+    /**
+   * Downloads the model corresponding to the source.
+   * @param source the source
+   */
+    IndexedDB.prototype.downloadModel = function (source) {
+        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
+            var src;
+            return (0, tslib_1.__generator)(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        src = this.infoFor(source);
+                        return [4 /*yield*/, __1.tf.io.copyModel(this.pathFor(source), "downloads://" + src.taskID + "_" + src.name)];
+                    case 1:
+                        _a.sent();
+                        return [2 /*return*/];
+                }
+            });
+        });
+    };
+    return IndexedDB;
+}(__1.Memory));
+exports.IndexedDB = IndexedDB;