def predict_label()

in services/github-bots/LabelBotFullFunctionality/LabelBot.py [0:0]

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

    def predict_label(self, issue_num):

        predict_issue = {"issues": [issue_num]}
        header = {"Content-Type": 'application/json'}
        response = requests.post(self.prediction_url, data=json.dumps(predict_issue), headers=header)
        predicted_labels = response.json()[0]["predictions"]

        if response.status_code == 200:
            logging.info(f'Successfully predicted labels to {issue_num}: {predicted_labels}')
        else:
            logging.error("Unable to predict labels")
            return False

        if 'Question' in predicted_labels:
            message = "Hey, this is the MXNet Label Bot and I think you have raised a question. \n" \
                      "For questions, you can also submit on MXNet discussion forum (https://discuss.mxnet.io), " \
                      "where it will get a wider audience and allow others to learn as well. Thanks! \n "
            self.add_github_labels(issue_num, ['question'])

        else:
            message = "Hey, this is the MXNet Label Bot. \n Thank you for submitting the issue! I will try and " \
                      "suggest some labels so that the appropriate MXNet community members can help " \
                      "resolve it. \n "
        if predicted_labels:
            message += 'Here are my recommended label(s): {}'.format(', '.join(predicted_labels))
        
        self.create_comment(issue_num, message)
        return True