def get_viable_instance_from_num_parameters()

in trending_deploy/models.py [0:0]

• 9 lines of code
• 3 McCabe index (conditional complexity)

def get_viable_instance_from_num_parameters(num_parameters: int):
    model_memory_usage_bytes = num_parameters * 4
    memory_factor = 2.2
    viable_instance = None
    for max_instance_memory_usage, instance in MEMORY_USAGE_TO_INSTANCE.items():
        if model_memory_usage_bytes * memory_factor < max_instance_memory_usage:
            viable_instance = instance
            break
    return viable_instance