@memlab/cli
Version:
command line interface for memlab
54 lines (53 loc) • 2.4 kB
JavaScript
;
/**
* Copyright (c) Meta Platforms, Inc. and affiliates.
*
* This source code is licensed under the MIT license found in the
* LICENSE file in the root directory of this source tree.
*
* @format
* @oncall memory_lab
*/
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const core_1 = require("@memlab/core");
const core_2 = require("@memlab/core");
const OptionConstant_1 = __importDefault(require("./lib/OptionConstant"));
class MLClusteringLinkageMaxDistanceOption extends core_2.BaseOption {
getOptionName() {
return OptionConstant_1.default.optionNames.ML_LINKAGE_MAX_DIST;
}
getDescription() {
return 'set linkage max distance value for clustering. The value should be between [0, 1] inclusive.';
}
parse(config, args) {
return __awaiter(this, void 0, void 0, function* () {
const name = this.getOptionName();
const arg = args[name];
if (arg) {
const linkageMaxDist = arg;
const linkageMaxDistNum = parseFloat(linkageMaxDist);
if (!isNaN(linkageMaxDistNum) &&
linkageMaxDistNum >= 0 &&
linkageMaxDistNum <= 1) {
config.mlClusteringLinkageMaxDistance = linkageMaxDistNum;
}
else {
core_1.utils.haltOrThrow(`ml-linkage-max-dist is not number between [0, 1]. ml-linkage-max-dist=${linkageMaxDist}`);
}
}
});
}
}
exports.default = MLClusteringLinkageMaxDistanceOption;