[ {
  "shortName" : "def model()",
  "longName" : "",
  "relativeFileName" : "src/model.py",
  "fileLinesCount" : 225,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 56,
  "mcCabeIndex" : 8,
  "numberOfParameters" : 6,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def attn()",
  "longName" : "",
  "relativeFileName" : "src/model.py",
  "fileLinesCount" : 225,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 40,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 6,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def main()",
  "longName" : "",
  "relativeFileName" : "src/run.py",
  "fileLinesCount" : 200,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 30,
  "mcCabeIndex" : 8,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def main()",
  "longName" : "",
  "relativeFileName" : "download.py",
  "fileLinesCount" : 66,
  "components" : [ "primary::ROOT" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 27,
  "mcCabeIndex" : 12,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def parse_arguments()",
  "longName" : "",
  "relativeFileName" : "src/run.py",
  "fileLinesCount" : 200,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 22,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def create_model()",
  "longName" : "",
  "relativeFileName" : "src/run.py",
  "fileLinesCount" : 200,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 22,
  "mcCabeIndex" : 4,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def iter_data()",
  "longName" : "",
  "relativeFileName" : "src/utils.py",
  "fileLinesCount" : 48,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 13,
  "mcCabeIndex" : 6,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def sample()",
  "longName" : "",
  "relativeFileName" : "src/run.py",
  "fileLinesCount" : 200,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 12,
  "mcCabeIndex" : 6,
  "numberOfParameters" : 9,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def parse_arguments()",
  "longName" : "",
  "relativeFileName" : "download.py",
  "fileLinesCount" : 66,
  "components" : [ "primary::ROOT" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 11,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def set_hparams()",
  "longName" : "",
  "relativeFileName" : "src/run.py",
  "fileLinesCount" : 200,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 11,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def reduce_mean()",
  "longName" : "",
  "relativeFileName" : "src/run.py",
  "fileLinesCount" : 200,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 11,
  "mcCabeIndex" : 2,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def count_parameters()",
  "longName" : "",
  "relativeFileName" : "src/utils.py",
  "fileLinesCount" : 48,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 9,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def load_data()",
  "longName" : "",
  "relativeFileName" : "src/run.py",
  "fileLinesCount" : 200,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 8,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def default_hparams()",
  "longName" : "",
  "relativeFileName" : "src/model.py",
  "fileLinesCount" : 225,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 8,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 0,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def norm()",
  "longName" : "",
  "relativeFileName" : "src/model.py",
  "fileLinesCount" : 225,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 8,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def block()",
  "longName" : "",
  "relativeFileName" : "src/model.py",
  "fileLinesCount" : 225,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 8,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def squared_euclidean_distance()",
  "longName" : "",
  "relativeFileName" : "src/utils.py",
  "fileLinesCount" : 48,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 7,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def evaluate()",
  "longName" : "",
  "relativeFileName" : "src/run.py",
  "fileLinesCount" : 200,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 6,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 11,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def conv1d()",
  "longName" : "",
  "relativeFileName" : "src/model.py",
  "fileLinesCount" : 225,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 6,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def mlp()",
  "longName" : "",
  "relativeFileName" : "src/model.py",
  "fileLinesCount" : 225,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 6,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 5,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def attention_mask()",
  "longName" : "",
  "relativeFileName" : "src/model.py",
  "fileLinesCount" : 225,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def color_quantize()",
  "longName" : "",
  "relativeFileName" : "src/utils.py",
  "fileLinesCount" : 48,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 5,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def set_seed()",
  "longName" : "",
  "relativeFileName" : "src/run.py",
  "fileLinesCount" : 200,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 4,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def shape_list()",
  "longName" : "",
  "relativeFileName" : "src/model.py",
  "fileLinesCount" : 225,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 4,
  "mcCabeIndex" : 3,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def softmax()",
  "longName" : "",
  "relativeFileName" : "src/model.py",
  "fileLinesCount" : 225,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 4,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def expand_tile()",
  "longName" : "",
  "relativeFileName" : "src/model.py",
  "fileLinesCount" : 225,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 4,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def positions_for()",
  "longName" : "",
  "relativeFileName" : "src/model.py",
  "fileLinesCount" : 225,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 4,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def split_states()",
  "longName" : "",
  "relativeFileName" : "src/model.py",
  "fileLinesCount" : 225,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 3,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 2,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def merge_states()",
  "longName" : "",
  "relativeFileName" : "src/model.py",
  "fileLinesCount" : 225,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 3,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def gelu()",
  "longName" : "",
  "relativeFileName" : "src/model.py",
  "fileLinesCount" : 225,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def gelu2()",
  "longName" : "",
  "relativeFileName" : "src/model.py",
  "fileLinesCount" : 225,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 1,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
}, {
  "shortName" : "def past_shape()",
  "longName" : "",
  "relativeFileName" : "src/model.py",
  "fileLinesCount" : 225,
  "components" : [ "primary::src" ],
  "startLine" : 0,
  "endLine" : 0,
  "linesOfCode" : 2,
  "mcCabeIndex" : 1,
  "numberOfParameters" : 4,
  "numberOfLiterals" : 0,
  "numberOfStatements" : 0,
  "numberOfExpressions" : 0
} ]