refactor: [EDXOLDMNG-224] code polishing for migrate cert config comm…

…and.
raccoongang · Dec 6, 2022 · 3ae9fcf · 3ae9fcf
1 parent 0d1162b
commit 3ae9fcf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cms/djangoapps/contentstore/management/commands/migrate_cert_config.py b/cms/djangoapps/contentstore/management/commands/migrate_cert_config.py
@@ -67,7 +67,7 @@ def _parse_course_key(self, raw_value: str) -> CourseKey:
 
         return result
 
-    def get_mongo_courses(self):
+    def get_mongo_courses(self) -> Iterator[CourseKey]:
         """
         Return objects for any mongo(old approach) modulestore backend course that has a certificate configuration.
         """
@@ -87,7 +87,7 @@ def get_mongo_courses(self):
                 course['_id']['name'],
             )
 
-    def get_split_courses(self):
+    def get_split_courses(self) -> Iterator[CourseKey]:
         """
         Return objects for any split modulestore backend course that has a certificate configuration.
         """