@epfml/discojs 2.0.0 → 2.1.2-p20240506085037.0

package/dist/memory/memory.js

@@ -1,160 +0,0 @@
-"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;

package/dist/{core/task → task}/data_example.d.ts

@@ -1,5 +1,5 @@
-export declare function isDataExample(raw: unknown): raw is DataExample;
 export interface DataExample {
     columnName: string;
     columnData: string | number;
 }
+export declare function isDataExample(raw: unknown): raw is DataExample;

package/dist/{core/task → task}/summary.d.ts

@@ -1,5 +1,5 @@
-export declare function isSummary(raw: unknown): raw is Summary;
 export interface Summary {
     preview: string;
     overview: string;
 }
+export declare function isSummary(raw: unknown): raw is Summary;