fixed data seeding issue (#2221)

Co-authored-by: Vignesh Hari <vichuhari100@gmail.com>
ohcnetwork · Jun 3, 2024 · a9f8d3d · a9f8d3d
1 parent 7b90428
commit a9f8d3d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/care/facility/management/commands/load_dummy_data.py b/care/facility/management/commands/load_dummy_data.py
@@ -30,5 +30,6 @@ def handle(self, *args, **options):
                 self.BASE_URL + "users.json",
                 self.BASE_URL + "facility.json",
             )
+            management.call_command("populate_investigations")
         except Exception as e:
             raise CommandError(e)
diff --git a/care/facility/models/patient_investigation.py b/care/facility/models/patient_investigation.py
@@ -29,7 +29,8 @@ class PatientInvestigation(BaseModel):
     choices = models.TextField(null=True, blank=True)
 
     def __str__(self) -> str:
-        return self.name + " in " + self.unit + " as " + self.investigation_type
+        unit_part = f" in {self.unit}" if self.unit else ""
+        return f"{self.name}{unit_part} as {self.investigation_type}"
 
 
 class InvestigationSession(BaseModel):