def lambda_handler()

in functions/start_model/app.py [0:0]


def lambda_handler(event, context):

    lookoutforvision_client = boto3.client("lookoutvision")
    project_name = os.environ["lookoutforvision_project_name"]
    project_model_version = os.environ["lookoutforvision_project_model_version"]

    try:
        running_states = ["HOSTED", "STARTING_HOSTING"]
        response = lookoutforvision_client.describe_model(
            ProjectName=project_name, ModelVersion=project_model_version
        )

        running_status = response["ModelDescription"]["Status"]

        if running_status not in running_states:
            response = lookoutforvision_client.start_model(
                ProjectName=project_name,
                ModelVersion=project_model_version,
                MinInferenceUnits=int(
                    os.environ["minimumInferenceUnitsToUse"]
                ),  # Can be increased upto 5 for running multiple inference units
                ClientToken=os.environ["clientToken"],
            )
            running_status = response["Status"]

        print("Current state is: ", running_status)
        return running_status

    except Exception as e:
        print(e)