Fix PR comments

mentoring/light_children.py

@@ -100,7 +100,7 @@ class LightChildrenMixin(XBlockWithChildrenFragmentsMixin):
     @classmethod
     def add_node_as_child(cls, block, xml_child, child_id):
         if xml_child.tag is etree.Comment:
-                return
+            return
         # Instantiate child
         child_class = cls.get_class_by_element(xml_child.tag)
         child = child_class(block)

mentoring/questionnaire.py

@@ -77,7 +77,7 @@ class QuestionnaireAbstractBlock(LightChild, StepMixin):
         name = self.__class__.__name__
 
         if str(self.type) not in self.valid_types:
-            raise ValueError('Invalid value for {}.type: `{}`'.format(name, self.type))
+            raise ValueError(u'Invalid value for {}.type: `{}`'.format(name, self.type))
 
         template_path = 'templates/html/{}_{}.html'.format(name.lower(), self.type)
         html = render_js_template(template_path, {

tests/integration/test_assessment.py

@@ -53,10 +53,9 @@ class MentoringAssessmentTest(MentoringBaseTest):
 
         @property
         def state(self):
-            state = {}
-            for choice in self._mcq.find_elements_by_css_selector(".choice"):
-                state[choice.text] = choice.find_element_by_css_selector("input").is_selected()
-            return state
+            return {
+                choice.text: choice.find_element_by_css_selector("input").is_selected()
+                for choice in self._mcq.find_elements_by_css_selector(".choice")}
 
         def select(self, text):
             state = {}