@gracexwho/model-card-generator/lib/python-program-analysis/src/specs/index.ts

Tool for generating model cards for Jupyter Notebook.

import * as builtins from "./__builtins__.json";
import * as random from "./random.json";
import * as matplotlib from "./matplotlib.json";
import * as pandas from "./pandas.json";
import * as sklearn from "./sklearn.json";
import * as numpy from "./numpy.json";

export interface FunctionSpec {
  name: string;
  updates?: (string | number)[];
  reads?: string[];
  returns?: string;
  returnsType?: TypeSpec<FunctionSpec>;
  higherorder?: number;
}

export type FunctionDescription = string | FunctionSpec;

export interface TypeSpec<FD> {
  methods?: FD[];
}

export interface ModuleSpec<FD> extends TypeSpec<FD> {
  functions?: FD[];
  modules?: ModuleMap<FD>;
  types?: { [typeName: string]: TypeSpec<FD> };
}

export interface ModuleMap<FD> {
  [moduleName: string]: ModuleSpec<FD>;
}

export type JsonSpecs = ModuleMap<FunctionDescription>;

export const DefaultSpecs: JsonSpecs = {
  ...builtins,
  ...random,
  ...matplotlib,
  ...pandas,
  ...sklearn,
  ...numpy
};