refactor: register unit models in app setup

Author: mariajgrimaldi

openedx_learning/apps/authoring/containers/apps.py

@@ -18,7 +18,7 @@ class ContainersConfig(AppConfig):
 
     def ready(self):
         """
-        Register Component and ComponentVersion.
+        Register ContainerEntity and ContainerEntityVersion.
         """
         from ..publishing.api import register_content_models  # pylint: disable=import-outside-toplevel
         from .models import ContainerEntity, ContainerEntityVersion  # pylint: disable=import-outside-toplevel

openedx_learning/apps/authoring/units/apps.py

@@ -14,3 +14,12 @@ class UnitsConfig(AppConfig):
     verbose_name = "Learning Core > Authoring > Units"
     default_auto_field = "django.db.models.BigAutoField"
     label = "oel_units"
+
+    def ready(self):
+        """
+        Register Unit and UnitVersion.
+        """
+        from ..publishing.api import register_content_models  # pylint: disable=import-outside-toplevel
+        from .models import Unit, UnitVersion  # pylint: disable=import-outside-toplevel
+
+        register_content_models(Unit, UnitVersion)