Function createModelResolver

sdk/dag-task-runner/src/dag.ts:209–217

(overrides: ModelMapOverride = {})

}

export function createModelResolver(overrides: ModelMapOverride = {}): (c: Complexity) => string {
  const models: ModelMap = { ...DEFAULT_MODEL_MAP, ...overrides };
  return (c: Complexity): string => {
    if (!COMPLEXITY_KEYS.includes(c)) {
      throw new Error(`Unknown complexity: ${c}`);
    }
    return models[c];
  };
}