official/nlp/modeling/ops/beam_search.py
File beam_search.py
has 562 lines of code (exceeds 250 allowed). Consider refactoring. Open
Open
# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
Function search
has a Cognitive Complexity of 18 (exceeds 5 allowed). Consider refactoring. Open
Open
def search(self, initial_ids, initial_cache, constraint_mask=None):
"""Beam search for sequences with highest scores.
Args:
initial_ids: initial ids to pass into the symbols_to_logits_fn. int tensor
- Read upRead up
Cognitive Complexity
Cognitive Complexity is a measure of how difficult a unit of code is to intuitively understand. Unlike Cyclomatic Complexity, which determines how difficult your code will be to test, Cognitive Complexity tells you how difficult your code will be to read and comprehend.
A method's cognitive complexity is based on a few simple rules:
- Code is not considered more complex when it uses shorthand that the language provides for collapsing multiple statements into one
- Code is considered more complex for each "break in the linear flow of the code"
- Code is considered more complex when "flow breaking structures are nested"
Further reading
Function sequence_beam_search
has 13 arguments (exceeds 4 allowed). Consider refactoring. Open
Open
def sequence_beam_search(
Function __init__
has 10 arguments (exceeds 4 allowed). Consider refactoring. Open
Open
def __init__(
Function _create_initial_state
has a Cognitive Complexity of 11 (exceeds 5 allowed). Consider refactoring. Open
Open
def _create_initial_state(
self, initial_ids, initial_cache, batch_size, constraint_mask=None
):
"""Return initial state dictionary and its shape invariants."""
for key, value in initial_cache.items():
- Read upRead up
Cognitive Complexity
Cognitive Complexity is a measure of how difficult a unit of code is to intuitively understand. Unlike Cyclomatic Complexity, which determines how difficult your code will be to test, Cognitive Complexity tells you how difficult your code will be to read and comprehend.
A method's cognitive complexity is based on a few simple rules:
- Code is not considered more complex when it uses shorthand that the language provides for collapsing multiple statements into one
- Code is considered more complex for each "break in the linear flow of the code"
- Code is considered more complex when "flow breaking structures are nested"