def int64_feature()

in basic_pitch/data/tf_example_serialization.py [0:0]

• 4 lines of code
• 2 McCabe index (conditional complexity)

def int64_feature(value: Union[List[int], int]) -> tf.train.Feature:
    if not isinstance(value, list):
        value = [value]
    return tf.train.Feature(int64_list=tf.train.Int64List(value=value))