def predict()

in right_size_your_sagemaker_endpoints/locust_file.py [0:0]

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

    def predict(self):
        input_file = "plants.jpg"
        with open(input_file, 'rb') as f:
            imageb64 = base64.b64encode(f.read())
        payload = {
            'data': imageb64.decode('utf-8'),
            'endpoint': os.environ['endpoint']
        }

        start_time = time.time()
        with self.client.post("", json=payload) as response:

            if type(response.json()) == dict:

                if response.json().get("body"):
                    total_time = int((time.time() - start_time) * 1000)
                    events.request_failure.fire(request_type="POST /dev/imageclassifier",
                                                name="MODELERROR",
                                                response_time=total_time,
                                                exception=response.json().get("body"),
                                                response_length=0)