def validate_input_shape()

in src/neo_loader/__init__.py [0:0]

• 11 lines of code
• 7 McCabe index (conditional complexity)

def validate_input_shape(framework, input_shape) -> {str: list}:
    if isinstance(input_shape, str):
        try:
            input_shape = json.loads(input_shape)
        except Exception as e:
            raise RuntimeError(f"InputConfiguration: Cannot load DataInputConfig. Incorrect json format. {e}")

    if not isinstance(input_shape, (dict, list)):
        raise RuntimeError("InputConfiguration: DataInputConfig is not dictionary or list.")

    if framework.lower() == Framework.PYTORCH.value and isinstance(input_shape, list):
        input_shape = {'input' + str(i): k for i, k in enumerate(input_shape)}
    return input_shape