def my_task()

in ezsmdeploy/data/smlocust.py [0:0]

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

    def my_task(self):
        start_time = time.time()
        try:
            if target_model == "":
                p.predict(input_data)
            else:
                p.predict(input_data, target_model=target_model)

            total_time = int((time.time() - start_time) * 1000)
            events.request_success.fire(
                request_type="sagemaker",
                name="predict",
                response_time=total_time,
                response_length=0,
            )

        except:
            total_time = int((time.time() - start_time) * 1000)
            events.request_failure.fire(
                request_type="sagemaker",
                name="predict",
                response_time=total_time,
                response_length=0,
                exception=sys.exc_info(),
            )