meta-edc 1.1.10__py3-none-any.whl → 1.1.13__py3-none-any.whl

meta_pharmacy/models/substitutions.py

@@ -42,7 +42,7 @@ class Substitutions(NonUniqueSubjectIdentifierFieldMixin, SiteModelMixin, BaseUu
     updated_visit_no = models.IntegerField("Visit Number", null=True, blank=True)
 
     updated_subject_identifier = models.CharField(
-        max_length=50, verbose_name="Taken from subject", null=True, blank=True
+        max_length=50, verbose_name="Taken from subject", default="", blank=True
     )
 
     arm_match = models.CharField(
@@ -82,7 +82,7 @@ class Substitutions(NonUniqueSubjectIdentifierFieldMixin, SiteModelMixin, BaseUu
     class Meta(SiteModelMixin.Meta, BaseUuidModel.Meta):
         verbose_name_plural = "IMP Substitutions"
         verbose_name = "IMP Substitution"
-        indexes = [
+        indexes = (
             Index(fields=["sid", "dispensed_sid", "subject_identifier"]),
             Index(fields=["row_index"]),
-        ]
+        )

meta_pharmacy/prepare_meta_pharmacy.py

@@ -9,7 +9,7 @@ style = color_style()
 
 
 def prepare_meta_pharmacy():
-    from django.apps import apps as django_apps
+    from django.apps import apps as django_apps  # noqa: PLC0415
 
     sys.stdout.write(style.MIGRATE_HEADING("Populating static pharmacy models for META3:\n"))
 

meta_pharmacy/utils/__init__.py

@@ -1 +1,3 @@
 from .update_initial_pharmacy_data import update_initial_pharmacy_data
+
+__all__ = ["update_initial_pharmacy_data"]

meta_pharmacy/utils/update_initial_pharmacy_data.py

@@ -136,7 +136,7 @@ def update_labels():
         LabelSpecification().save()
         default = LabelSpecification.objects.get(name="default")
 
-    for name, label_config in site_label_configs.registry.items():
+    for name in site_label_configs.registry:
         LabelConfiguration.objects.create(name=name, label_specification=default)
     for label_configuration in LabelConfiguration.objects.filter(name__contains="patient"):
         label_configuration.requires_allocation = True

meta_prn/admin/__init__.py

@@ -11,3 +11,19 @@ from .onschedule_dm_referral_admin import OnScheduleDmReferralAdmin
 from .pregnancy_notification_admin import PregnancyNotificationAdmin
 from .protocol_incident_admin import ProtocolIncidentAdmin
 from .subject_transfer_admin import SubjectTransferAdmin
+
+__all__ = [
+    "DmReferralAdmin",
+    "EndOfStudyAdmin",
+    "LossToFollowupAdmin",
+    "OffScheduleAdmin",
+    "OffScheduleDmReferralAdmin",
+    "OffSchedulePostnatalAdmin",
+    "OffSchedulePregnancyAdmin",
+    "OffStudyMedicationAdmin",
+    "OnScheduleAdmin",
+    "OnScheduleDmReferralAdmin",
+    "PregnancyNotificationAdmin",
+    "ProtocolIncidentAdmin",
+    "SubjectTransferAdmin",
+]

meta_prn/admin/dm_referral_admin.py

@@ -1,6 +1,7 @@
 from django.contrib import admin
 from django_audit_fields.admin import audit_fieldset_tuple
-from edc_action_item import ActionItemModelAdminMixin, action_fieldset_tuple
+from edc_action_item.fieldsets import action_fieldset_tuple
+from edc_action_item.modeladmin_mixins import ActionItemModelAdminMixin
 from edc_model_admin.dashboard import ModelAdminSubjectDashboardMixin
 from edc_model_admin.history import SimpleHistoryAdmin
 from edc_sites.admin import SiteModelAdminMixin

meta_prn/admin/end_of_study_admin.py

@@ -1,12 +1,11 @@
-from typing import Tuple
-
 from dateutil.relativedelta import relativedelta
 from django.contrib import admin
 from django.template.loader import render_to_string
 from django.utils.html import format_html
 from django.utils.safestring import mark_safe
 from django_audit_fields.admin import audit_fieldset_tuple
-from edc_action_item import ActionItemModelAdminMixin, action_fieldset_tuple
+from edc_action_item.fieldsets import action_fieldset_tuple
+from edc_action_item.modeladmin_mixins import ActionItemModelAdminMixin
 from edc_constants.constants import OTHER
 from edc_model_admin.dashboard import ModelAdminSubjectDashboardMixin
 from edc_model_admin.history import SimpleHistoryAdmin
@@ -30,7 +29,7 @@ class EndOfStudyAdmin(
 ):
     additional_instructions = format_html(
         "{html}",
-        html=mark_safe(
+        html=mark_safe(  # noqa: S308
             render_to_string(
                 "meta_prn/eos/additional_instructions.html",
                 context=dict(
@@ -123,7 +122,7 @@ class EndOfStudyAdmin(
             kwargs["queryset"] = OffstudyReasons.objects.order_by("display_index")
         return super().formfield_for_foreignkey(db_field, request, **kwargs)
 
-    def get_list_display(self, request) -> Tuple[str, ...]:
+    def get_list_display(self, request) -> tuple[str, ...]:
         list_display = super().get_list_display(request)
         custom_fields = (
             "subject_identifier",
@@ -137,12 +136,12 @@ class EndOfStudyAdmin(
             f for f in list_display if f not in custom_fields + ("__str__",)
         )
 
-    def get_list_filter(self, request) -> Tuple[str, ...]:
+    def get_list_filter(self, request) -> tuple[str, ...]:
         list_filter = super().get_list_filter(request)
         custom_fields = ("offstudy_reason", "last_seen_date")
         return custom_fields + tuple(f for f in list_filter if f not in custom_fields)
 
-    def get_search_fields(self, request) -> Tuple[str, ...]:
+    def get_search_fields(self, request) -> tuple[str, ...]:
         search_fields = super().get_search_fields(request)
         custom_fields = ("subject_identifier",)
         return tuple(set(custom_fields + search_fields))

meta_prn/admin/loss_to_followup_admin.py

@@ -1,6 +1,7 @@
 from django.contrib import admin
 from django_audit_fields.admin import audit_fieldset_tuple
-from edc_action_item import ActionItemModelAdminMixin, action_fieldset_tuple
+from edc_action_item.fieldsets import action_fieldset_tuple
+from edc_action_item.modeladmin_mixins import ActionItemModelAdminMixin
 from edc_model_admin.dashboard import ModelAdminSubjectDashboardMixin
 from edc_model_admin.history import SimpleHistoryAdmin
 from edc_sites.admin import SiteModelAdminMixin
@@ -54,7 +55,7 @@ class LossToFollowupAdmin(
         "number_consecutive_missed_visits",
     )
 
-    radio_fields = {
+    radio_fields = {  # noqa: RUF012
         "home_visited": admin.VERTICAL,
         "loss_category": admin.VERTICAL,
     }

meta_prn/admin/off_study_medication_admin.py

@@ -1,8 +1,7 @@
-from typing import Tuple
-
 from django.contrib import admin
 from django_audit_fields.admin import audit_fieldset_tuple
-from edc_action_item import ActionItemModelAdminMixin, action_fieldset_tuple
+from edc_action_item.fieldsets import action_fieldset_tuple
+from edc_action_item.modeladmin_mixins import ActionItemModelAdminMixin
 from edc_model_admin.dashboard import ModelAdminSubjectDashboardMixin
 from edc_model_admin.history import SimpleHistoryAdmin
 from edc_sites.admin import SiteModelAdminMixin
@@ -40,7 +39,7 @@ class OffStudyMedicationAdmin(
         audit_fieldset_tuple,
     )
 
-    radio_fields = {
+    radio_fields = {  # noqa: RUF012
         "reason": admin.VERTICAL,
     }
 
@@ -48,7 +47,7 @@ class OffStudyMedicationAdmin(
 
     search_fields = ("subject_identifier",)
 
-    def get_list_display(self, request) -> Tuple[str, ...]:
+    def get_list_display(self, request) -> tuple[str, ...]:
         list_display = super().get_list_display(request)
         custom_fields = (
             "subject_identifier",
@@ -58,7 +57,7 @@ class OffStudyMedicationAdmin(
         )
         return custom_fields + tuple(f for f in list_display if f not in custom_fields)
 
-    def get_list_filter(self, request) -> Tuple[str, ...]:
+    def get_list_filter(self, request) -> tuple[str, ...]:
         list_filter = super().get_list_filter(request)
         custom_fields = (
             "medications",

meta_prn/admin/offschedule_admin.py

@@ -1,10 +1,9 @@
-from typing import Tuple
-
 from django.contrib import admin
 from django.template.loader import render_to_string
 from django.utils.safestring import mark_safe
 from django_audit_fields import audit_fieldset_tuple
-from edc_action_item import ActionItemModelAdminMixin, action_fieldset_tuple
+from edc_action_item.fieldsets import action_fieldset_tuple
+from edc_action_item.modeladmin_mixins import ActionItemModelAdminMixin
 from edc_model_admin.dashboard import ModelAdminSubjectDashboardMixin
 from edc_model_admin.history import SimpleHistoryAdmin
 from edc_sites.admin import SiteModelAdminMixin
@@ -23,7 +22,7 @@ class OffScheduleAdmin(
 ):
     form = OffScheduleForm
 
-    additional_instructions = mark_safe(
+    additional_instructions = mark_safe(  # noqa: S308
         render_to_string(
             "meta_prn/offschedule/additional_instructions.html",
             context=dict(verbose_name=EndOfStudy._meta.verbose_name),
@@ -36,12 +35,12 @@ class OffScheduleAdmin(
         audit_fieldset_tuple,
     )
 
-    def get_list_display(self, request) -> Tuple[str, ...]:
+    def get_list_display(self, request) -> tuple[str, ...]:
         list_display = super().get_list_display(request)
         custom_fields = ("subject_identifier", "dashboard", "offschedule_datetime")
         return custom_fields + tuple(f for f in list_display if f not in custom_fields)
 
-    def get_list_filter(self, request) -> Tuple[str, ...]:
+    def get_list_filter(self, request) -> tuple[str, ...]:
         list_filter = super().get_list_filter(request)
         custom_fields = ("offschedule_datetime",)
         return custom_fields + tuple(f for f in list_filter if f not in custom_fields)

meta_prn/admin/offschedule_dm_referral_admin.py

@@ -1,10 +1,10 @@
-from typing import Tuple
 
 from django.contrib import admin
 from django.template.loader import render_to_string
 from django.utils.safestring import mark_safe
 from django_audit_fields import audit_fieldset_tuple
-from edc_action_item import ActionItemModelAdminMixin, action_fieldset_tuple
+from edc_action_item.fieldsets import action_fieldset_tuple
+from edc_action_item.modeladmin_mixins import ActionItemModelAdminMixin
 from edc_model_admin.dashboard import ModelAdminSubjectDashboardMixin
 from edc_model_admin.history import SimpleHistoryAdmin
 from edc_sites.admin import SiteModelAdminMixin
@@ -25,12 +25,12 @@ class OffScheduleDmReferralAdmin(
 
     form = OffScheduleDmReferralForm
 
-    additional_instructions = mark_safe(
+    additional_instructions = mark_safe(  # noqa: S308
         render_to_string(
             "meta_prn/offschedule/additional_instructions.html",
             context=dict(verbose_name=EndOfStudy._meta.verbose_name),
         )
-    )  # nosec B703, B308
+    )
 
     fieldsets = (
         (None, {"fields": ("subject_identifier", "offschedule_datetime")}),
@@ -38,12 +38,12 @@ class OffScheduleDmReferralAdmin(
         audit_fieldset_tuple,
     )
 
-    def get_list_display(self, request) -> Tuple[str, ...]:
+    def get_list_display(self, request) -> tuple[str, ...]:
         list_display = super().get_list_display(request)
         custom_fields = ("subject_identifier", "dashboard", "offschedule_datetime")
         return custom_fields + tuple(f for f in list_display if f not in custom_fields)
 
-    def get_list_filter(self, request) -> Tuple[str, ...]:
+    def get_list_filter(self, request) -> tuple[str, ...]:
         list_filter = super().get_list_filter(request)
         custom_fields = ("offschedule_datetime",)
         return custom_fields + tuple(f for f in list_filter if f not in custom_fields)

meta_prn/admin/offschedule_postnatal_admin.py

@@ -1,10 +1,10 @@
-from typing import Tuple
 
 from django.contrib import admin
 from django.template.loader import render_to_string
 from django.utils.safestring import mark_safe
 from django_audit_fields import audit_fieldset_tuple
-from edc_action_item import ActionItemModelAdminMixin, action_fieldset_tuple
+from edc_action_item.fieldsets import action_fieldset_tuple
+from edc_action_item.modeladmin_mixins import ActionItemModelAdminMixin
 from edc_model_admin.dashboard import ModelAdminSubjectDashboardMixin
 from edc_model_admin.history import SimpleHistoryAdmin
 from edc_sites.admin import SiteModelAdminMixin
@@ -22,12 +22,12 @@ class OffSchedulePostnatalAdmin(
 ):
     instructions = None
 
-    additional_instructions = mark_safe(
+    additional_instructions = mark_safe(  # noqa: S308
         render_to_string(
             "meta_prn/offschedule/additional_instructions.html",
             context=dict(verbose_name=EndOfStudy._meta.verbose_name),
         )
-    )  # nosec B703, B308
+    )
 
     fieldsets = (
         (None, {"fields": ("subject_identifier", "offschedule_datetime")}),
@@ -35,16 +35,16 @@ class OffSchedulePostnatalAdmin(
         audit_fieldset_tuple,
     )
 
-    def get_list_display(self, request) -> Tuple[str, ...]:
+    def get_list_display(self, request) -> tuple[str, ...]:
         list_display = super().get_list_display(request)
         custom_fields = ("subject_identifier", "dashboard", "offschedule_datetime")
         return custom_fields + tuple(f for f in list_display if f not in custom_fields)
 
-    def get_list_filter(self, request) -> Tuple[str, ...]:
+    def get_list_filter(self, request) -> tuple[str, ...]:
         list_filter = super().get_list_filter(request)
         custom_fields = ("offschedule_datetime",)
         return custom_fields + tuple(f for f in list_filter if f not in custom_fields)
 
-    def get_readonly_fields(self, request, obj=None) -> Tuple[str, ...]:
+    def get_readonly_fields(self, request, obj=None) -> tuple[str, ...]:
         custom_fields = ("subject_identifier", "offschedule_datetime")
         return tuple(set(super().get_readonly_fields(request, obj=obj) + custom_fields))

meta_prn/admin/offschedule_pregnancy_admin.py

@@ -1,10 +1,10 @@
-from typing import Tuple
 
 from django.contrib import admin
 from django.template.loader import render_to_string
 from django.utils.safestring import mark_safe
 from django_audit_fields import audit_fieldset_tuple
-from edc_action_item import ActionItemModelAdminMixin, action_fieldset_tuple
+from edc_action_item.fieldsets import action_fieldset_tuple
+from edc_action_item.modeladmin_mixins import ActionItemModelAdminMixin
 from edc_model_admin.dashboard import ModelAdminSubjectDashboardMixin
 from edc_model_admin.history import SimpleHistoryAdmin
 from edc_sites.admin import SiteModelAdminMixin
@@ -23,12 +23,12 @@ class OffSchedulePregnancyAdmin(
 ):
     instructions = None
 
-    additional_instructions = mark_safe(
+    additional_instructions = mark_safe(  # noqa: S308
         render_to_string(
             "meta_prn/offschedule/additional_instructions.html",
             context=dict(verbose_name=EndOfStudy._meta.verbose_name),
         )
-    )  # nosec B703, B308
+    )
 
     form = OffSchedulePregnancyForm
 
@@ -38,16 +38,17 @@ class OffSchedulePregnancyAdmin(
         audit_fieldset_tuple,
     )
 
-    def get_list_display(self, request) -> Tuple[str, ...]:
+    def get_list_display(self, request) -> tuple[str, ...]:
         list_display = super().get_list_display(request)
         custom_fields = ("subject_identifier", "dashboard", "offschedule_datetime")
         return custom_fields + tuple(f for f in list_display if f not in custom_fields)
 
-    def get_list_filter(self, request) -> Tuple[str, ...]:
+    def get_list_filter(self, request) -> tuple[str, ...]:
         list_filter = super().get_list_filter(request)
         custom_fields = ("offschedule_datetime",)
         return custom_fields + tuple(f for f in list_filter if f not in custom_fields)
 
     # def get_readonly_fields(self, request, obj=None) -> Tuple[str, ...]:
     #     custom_fields = ("subject_identifier", "offschedule_datetime")
-    #     return tuple(set(super().get_readonly_fields(request, obj=obj) + custom_fields))
+    #     return tuple(set(super().get_readonly_fields(request, obj=obj) +
+    #     custom_fields))

meta_prn/admin/onschedule_admin.py

@@ -1,5 +1,3 @@
-from typing import Tuple
-
 from django.contrib import admin
 from edc_model_admin.dashboard import ModelAdminSubjectDashboardMixin
 from edc_model_admin.history import SimpleHistoryAdmin
@@ -17,23 +15,24 @@ class OnScheduleAdmin(
 
     fields = ("subject_identifier", "onschedule_datetime")
 
-    def get_list_display(self, request) -> Tuple[str, ...]:
+    def get_list_display(self, request) -> tuple[str, ...]:
         list_display = super().get_list_display(request)
         custom_fields = ("subject_identifier", "dashboard", "onschedule_datetime")
         return custom_fields + tuple(
-            f for f in list_display if f not in custom_fields + ("__str__",)
+            f for f in list_display if f not in (*custom_fields, "__str__")
         )
 
-    def get_list_filter(self, request) -> Tuple[str, ...]:
+    def get_list_filter(self, request) -> tuple[str, ...]:
         list_filter = super().get_list_filter(request)
         custom_fields = ("onschedule_datetime",)
         return custom_fields + tuple(f for f in list_filter if f not in custom_fields)
 
-    def get_readonly_fields(self, request, obj=None) -> Tuple[str, ...]:
+    def get_readonly_fields(self, request, obj=None) -> tuple[str, ...]:  # noqa: ARG002
         return (
             "subject_identifier",
             "onschedule_datetime",
         )
 
-    def get_search_fields(self, request) -> Tuple[str, ...]:
-        return tuple(set(super().get_search_fields(request) + ("subject_identifier",)))
+    def get_search_fields(self, request) -> tuple[str, ...]:
+        fields = super().get_search_fields(request)
+        return tuple({*fields, "subject_identifier"})

meta_prn/admin/onschedule_dm_referral_admin.py

@@ -1,5 +1,3 @@
-from typing import Tuple
-
 from django.contrib import admin
 from edc_model_admin.dashboard import ModelAdminSubjectDashboardMixin
 from edc_model_admin.history import SimpleHistoryAdmin
@@ -17,14 +15,14 @@ class OnScheduleDmReferralAdmin(
 
     fields = ("subject_identifier", "onschedule_datetime")
 
-    def get_list_display(self, request) -> Tuple[str, ...]:
+    def get_list_display(self, request) -> tuple[str, ...]:
         list_display = super().get_list_display(request)
         custom_fields = ("subject_identifier", "dashboard", "onschedule_datetime")
         return custom_fields + tuple(
-            f for f in list_display if f not in custom_fields + ("__str__",)
+            f for f in list_display if f not in (*custom_fields, "__str__")
         )
 
-    def get_list_filter(self, request) -> Tuple[str, ...]:
+    def get_list_filter(self, request) -> tuple[str, ...]:
         list_filter = super().get_list_filter(request)
         custom_fields = ("onschedule_datetime",)
         return custom_fields + tuple(f for f in list_filter if f not in custom_fields)
@@ -35,5 +33,6 @@ class OnScheduleDmReferralAdmin(
     #         "onschedule_datetime",
     #     )
 
-    def get_search_fields(self, request) -> Tuple[str, ...]:
-        return tuple(set(super().get_search_fields(request) + ("subject_identifier",)))
+    def get_search_fields(self, request) -> tuple[str, ...]:
+        fields = super().get_search_fields(request)
+        return tuple({*fields, "subject_identifier"})

meta_prn/admin/pregnancy_notification_admin.py

@@ -1,8 +1,7 @@
-from typing import Tuple
-
 from django.contrib import admin
 from django_audit_fields.admin import audit_fieldset_tuple
-from edc_action_item import ActionItemModelAdminMixin, action_fieldset_tuple
+from edc_action_item.fieldsets import action_fieldset_tuple
+from edc_action_item.modeladmin_mixins import ActionItemModelAdminMixin
 from edc_model_admin.dashboard import ModelAdminSubjectDashboardMixin
 from edc_model_admin.history import SimpleHistoryAdmin
 from edc_sites.admin import SiteModelAdminMixin
@@ -46,12 +45,12 @@ class PregnancyNotificationAdmin(
         audit_fieldset_tuple,
     )
 
-    radio_fields = {
+    radio_fields = {  # noqa: RUF012
         "bhcg_confirmed": admin.VERTICAL,
         "may_contact": admin.VERTICAL,
     }
 
-    def get_list_display(self, request) -> Tuple[str, ...]:
+    def get_list_display(self, request) -> tuple[str, ...]:
         list_display = super().get_list_display(request)
         custom_fields = (
             "subject_identifier",
@@ -61,7 +60,7 @@ class PregnancyNotificationAdmin(
         )
         return custom_fields + tuple(f for f in list_display if f not in custom_fields)
 
-    def get_list_filter(self, request) -> Tuple[str, ...]:
+    def get_list_filter(self, request) -> tuple[str, ...]:
         list_filter = super().get_list_filter(request)
         custom_fields = ("edd", "may_contact")
         return custom_fields + tuple(f for f in list_filter if f not in custom_fields)

meta_prn/admin/protocol_incident_admin.py

@@ -1,5 +1,5 @@
 from django.contrib import admin
-from edc_action_item import ActionItemModelAdminMixin
+from edc_action_item.modeladmin_mixins import ActionItemModelAdminMixin
 from edc_model_admin.dashboard import ModelAdminSubjectDashboardMixin
 from edc_model_admin.history import SimpleHistoryAdmin
 from edc_protocol_incident.modeladmin_mixins import ProtocolIncidentModelAdminMixin

meta_prn/admin/subject_transfer_admin.py

@@ -1,5 +1,5 @@
 from django.contrib import admin
-from edc_action_item import ActionItemModelAdminMixin
+from edc_action_item.modeladmin_mixins import ActionItemModelAdminMixin
 from edc_model_admin.dashboard import ModelAdminSubjectDashboardMixin
 from edc_model_admin.history import SimpleHistoryAdmin
 from edc_sites.admin import SiteModelAdminMixin

meta_prn/baker_recipes.py

@@ -1,6 +1,6 @@
 from dateutil.relativedelta import relativedelta
+from django.utils import timezone
 from edc_constants.constants import YES
-from edc_utils import get_utcnow
 from faker import Faker
 from model_bakery.recipe import Recipe
 
@@ -14,15 +14,15 @@ pregnancynotification = Recipe(
     # site=None,
     action_identifier=None,
     subject_identifier=None,
-    report_datetime=get_utcnow(),
+    report_datetime=timezone.now(),
     bhcg_confirmed=YES,
     unconfirmed_details=None,
-    edd=get_utcnow() + relativedelta(months=6),
+    edd=timezone.now() + relativedelta(months=6),
 )
 
 dmreferral = Recipe(
     DmReferral,
     action_identifier=None,
     subject_identifier=None,
-    report_datetime=get_utcnow(),
+    report_datetime=timezone.now(),
 )

meta_prn/form_validators/__init__.py

@@ -1,2 +1,7 @@
 from .end_of_study import EndOfStudyFormValidator
 from .protocol_incident import ProtocolIncidentFormValidator
+
+__all__ = [
+    "EndOfStudyFormValidator",
+    "ProtocolIncidentFormValidator",
+]

meta_prn/form_validators/end_of_study.py

@@ -192,7 +192,7 @@ class EndOfStudyFormValidator(
                 if (obj.bhcg_date or obj.report_datetime.date()) != self.cleaned_data.get(
                     "pregnancy_date"
                 ):
-                    dt = formatted_date((obj.bhcg_date or obj.report_datetime.date()))
+                    dt = formatted_date(obj.bhcg_date or obj.report_datetime.date())
                     self.raise_validation_error(
                         {
                             "pregnancy_date": (
@@ -229,7 +229,7 @@ class EndOfStudyFormValidator(
                     obj.delivery_datetime.date() or obj.report_datetime.date()
                 ) != self.cleaned_data.get("delivery_date"):
                     dt = formatted_date(
-                        (obj.delivery_datetime.date() or obj.report_datetime.date())
+                        obj.delivery_datetime.date() or obj.report_datetime.date()
                     )
                     self.raise_validation_error(
                         {

meta_prn/forms/__init__.py

@@ -8,3 +8,16 @@ from .offschedule_pregnancy_form import OffSchedulePregnancyForm
 from .pregnancy_notification_form import PregnancyNotificationForm
 from .protocol_incident_form import ProtocolIncidentForm
 from .subject_transfer_form import SubjectTransferForm
+
+__all__ = [
+    "DmReferralForm",
+    "EndOfStudyForm",
+    "LossToFollowupForm",
+    "OffScheduleDmReferralForm",
+    "OffScheduleForm",
+    "OffSchedulePregnancyForm",
+    "OffStudyMedicationForm",
+    "PregnancyNotificationForm",
+    "ProtocolIncidentForm",
+    "SubjectTransferForm",
+]