Move data_generation to sandbox

jenniferjiangkells · jenniferjiangkells · commit 869d4de471dc · 2025-10-28T17:21:12.000Z
diff --git a/healthchain/__init__.py b/healthchain/__init__.py
@@ -16,3 +16,18 @@
 
 # Export them at the top level
 __all__ = ["ConfigManager", "ValidationLevel", "api", "ehr", "sandbox"]
+
+
+# Legacy import with warning
+def __getattr__(name):
+    if name == "data_generators":
+        warnings.warn(
+            "Importing data_generators from healthchain is deprecated. "
+            "Use 'from healthchain.sandbox import generators' instead.",
+            DeprecationWarning,
+            stacklevel=2,
+        )
+        from healthchain.sandbox import generators
+
+        return generators
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
diff --git a/healthchain/sandbox/decorator.py b/healthchain/sandbox/decorator.py
@@ -106,7 +106,7 @@ def decorator(func: F) -> F:
         @wraps(func)
         def wrapper(self, *args: Any, **kwargs: Any) -> Any:
             # Import here to avoid circular imports
-            from healthchain.data_generators import CdsDataGenerator
+            from healthchain.sandbox.generators import CdsDataGenerator
             from healthchain.sandbox.clients.ehr import EHRClient
 
             # Validate function decorated is a use case base class
diff --git a/healthchain/sandbox/generators/__init__.py b/healthchain/sandbox/generators/__init__.py
diff --git a/healthchain/sandbox/generators/basegenerators.py b/healthchain/sandbox/generators/basegenerators.py
diff --git a/healthchain/sandbox/generators/cdsdatagenerator.py b/healthchain/sandbox/generators/cdsdatagenerator.py
@@ -7,7 +7,7 @@
 
 from fhir.resources.resource import Resource
 
-from healthchain.data_generators.basegenerators import generator_registry
+from healthchain.sandbox.generators.basegenerators import generator_registry
 from healthchain.models import Prefetch
 from healthchain.fhir import create_document_reference
 from healthchain.sandbox.workflows import Workflow
diff --git a/healthchain/sandbox/generators/conditiongenerators.py b/healthchain/sandbox/generators/conditiongenerators.py
@@ -5,13 +5,13 @@
 from fhir.resources.condition import ConditionStage, ConditionParticipant
 
 from healthchain.fhir.helpers import create_single_codeable_concept, create_condition
-from healthchain.data_generators.basegenerators import (
+from healthchain.sandbox.generators.basegenerators import (
     BaseGenerator,
     generator_registry,
     register_generator,
     CodeableConceptGenerator,
 )
-from healthchain.data_generators.value_sets.conditioncodes import (
+from healthchain.sandbox.generators.value_sets.conditioncodes import (
     ConditionCodeSimple,
     ConditionCodeComplex,
 )
diff --git a/healthchain/sandbox/generators/encountergenerators.py b/healthchain/sandbox/generators/encountergenerators.py
@@ -7,7 +7,7 @@
 from fhir.resources.codeableconcept import CodeableConcept
 from fhir.resources.period import Period
 from fhir.resources.reference import Reference
-from healthchain.data_generators.basegenerators import (
+from healthchain.sandbox.generators.basegenerators import (
     BaseGenerator,
     generator_registry,
     register_generator,
diff --git a/healthchain/sandbox/generators/generator_templates/templates.py b/healthchain/sandbox/generators/generator_templates/templates.py
diff --git a/healthchain/sandbox/generators/medicationadministrationgenerators.py b/healthchain/sandbox/generators/medicationadministrationgenerators.py
@@ -5,7 +5,7 @@
 from fhir.resources.medicationadministration import MedicationAdministrationDosage
 from fhir.resources.reference import Reference
 from fhir.resources.codeablereference import CodeableReference
-from healthchain.data_generators.basegenerators import (
+from healthchain.sandbox.generators.basegenerators import (
     BaseGenerator,
     generator_registry,
     register_generator,
diff --git a/healthchain/sandbox/generators/medicationrequestgenerators.py b/healthchain/sandbox/generators/medicationrequestgenerators.py
@@ -1,13 +1,13 @@
 from typing import Optional
 from faker import Faker
 
-from healthchain.data_generators.basegenerators import (
+from healthchain.sandbox.generators.basegenerators import (
     BaseGenerator,
     generator_registry,
     register_generator,
     CodeableConceptGenerator,
 )
-from healthchain.data_generators.value_sets.medicationcodes import (
+from healthchain.sandbox.generators.value_sets.medicationcodes import (
     MedicationRequestMedication,
 )
 from fhir.resources.medicationrequest import MedicationRequest
diff --git a/healthchain/sandbox/generators/patientgenerators.py b/healthchain/sandbox/generators/patientgenerators.py
@@ -1,7 +1,7 @@
 from typing import Optional
 from faker import Faker
 
-from healthchain.data_generators.basegenerators import (
+from healthchain.sandbox.generators.basegenerators import (
     BaseGenerator,
     generator_registry,
     register_generator,
diff --git a/healthchain/sandbox/generators/practitionergenerators.py b/healthchain/sandbox/generators/practitionergenerators.py
@@ -1,7 +1,7 @@
 from typing import Optional
 from faker import Faker
 
-from healthchain.data_generators.basegenerators import (
+from healthchain.sandbox.generators.basegenerators import (
     BaseGenerator,
     generator_registry,
     register_generator,
diff --git a/healthchain/sandbox/generators/proceduregenerators.py b/healthchain/sandbox/generators/proceduregenerators.py
@@ -1,13 +1,13 @@
 from typing import Optional
 from faker import Faker
 
-from healthchain.data_generators.basegenerators import (
+from healthchain.sandbox.generators.basegenerators import (
     BaseGenerator,
     generator_registry,
     register_generator,
     CodeableConceptGenerator,
 )
-from healthchain.data_generators.value_sets.procedurecodes import (
+from healthchain.sandbox.generators.value_sets.procedurecodes import (
     ProcedureCodeSimple,
     ProcedureCodeComplex,
 )
diff --git a/healthchain/sandbox/generators/value_sets/__init__.py b/healthchain/sandbox/generators/value_sets/__init__.py
diff --git a/healthchain/sandbox/generators/value_sets/base.py b/healthchain/sandbox/generators/value_sets/base.py
diff --git a/healthchain/sandbox/generators/value_sets/codesystems.py b/healthchain/sandbox/generators/value_sets/codesystems.py
diff --git a/healthchain/sandbox/generators/value_sets/conditioncodes.py b/healthchain/sandbox/generators/value_sets/conditioncodes.py
diff --git a/healthchain/sandbox/generators/value_sets/medicationcodes.py b/healthchain/sandbox/generators/value_sets/medicationcodes.py
diff --git a/healthchain/sandbox/generators/value_sets/procedurecodes.py b/healthchain/sandbox/generators/value_sets/procedurecodes.py
diff --git a/healthchain/sandbox/sandbox_client.py b/healthchain/sandbox/sandbox_client.py
@@ -304,7 +304,7 @@ def load_free_text(
             ...     random_seed=42
             ... )
         """
-        from healthchain.data_generators import CdsDataGenerator
+        from .generators import CdsDataGenerator
 
         workflow_enum = Workflow(workflow) if isinstance(workflow, str) else workflow
 
diff --git a/tests/sandbox/generators/test_cds_data_generator.py b/tests/sandbox/generators/test_cds_data_generator.py
@@ -5,7 +5,7 @@
 from fhir.resources.procedure import Procedure
 from fhir.resources.patient import Patient
 
-from healthchain.data_generators import CdsDataGenerator
+from healthchain.sandbox.generators import CdsDataGenerator
 from healthchain.sandbox.workflows import Workflow
 
 
diff --git a/tests/sandbox/generators/test_condition_generators.py b/tests/sandbox/generators/test_condition_generators.py
@@ -1,10 +1,10 @@
-from healthchain.data_generators.conditiongenerators import (
+from healthchain.sandbox.generators.conditiongenerators import (
     ClinicalStatusGenerator,
     VerificationStatusGenerator,
     CategoryGenerator,
     ConditionGenerator,
 )
-from healthchain.data_generators.value_sets.conditioncodes import (
+from healthchain.sandbox.generators.value_sets.conditioncodes import (
     ConditionCodeSimple,
     ConditionCodeComplex,
 )
diff --git a/tests/sandbox/generators/test_encounter_generators.py b/tests/sandbox/generators/test_encounter_generators.py
@@ -1,4 +1,4 @@
-from healthchain.data_generators.encountergenerators import (
+from healthchain.sandbox.generators.encountergenerators import (
     ClassGenerator,
     EncounterTypeGenerator,
     EncounterGenerator,
diff --git a/tests/sandbox/generators/test_medication_administration_generators.py b/tests/sandbox/generators/test_medication_administration_generators.py
@@ -1,8 +1,8 @@
-from healthchain.data_generators.medicationadministrationgenerators import (
+from healthchain.sandbox.generators.medicationadministrationgenerators import (
     MedicationAdministrationDosageGenerator,
     MedicationAdministrationGenerator,
 )
-from healthchain.data_generators.value_sets.medicationcodes import (
+from healthchain.sandbox.generators.value_sets.medicationcodes import (
     MedicationRequestMedication,
 )
 
diff --git a/tests/sandbox/generators/test_medication_request_generators.py b/tests/sandbox/generators/test_medication_request_generators.py
@@ -1,8 +1,8 @@
-from healthchain.data_generators.medicationrequestgenerators import (
+from healthchain.sandbox.generators.medicationrequestgenerators import (
     MedicationRequestGenerator,
     MedicationRequestContainedGenerator,
 )
-from healthchain.data_generators.value_sets.medicationcodes import (
+from healthchain.sandbox.generators.value_sets.medicationcodes import (
     MedicationRequestMedication,
 )
 
diff --git a/tests/sandbox/generators/test_patient_generators.py b/tests/sandbox/generators/test_patient_generators.py
@@ -1,4 +1,4 @@
-from healthchain.data_generators.patientgenerators import (
+from healthchain.sandbox.generators.patientgenerators import (
     PatientGenerator,
     HumanNameGenerator,
 )
diff --git a/tests/sandbox/generators/test_practitioner_generators.py b/tests/sandbox/generators/test_practitioner_generators.py
@@ -1,4 +1,4 @@
-from healthchain.data_generators.practitionergenerators import (
+from healthchain.sandbox.generators.practitionergenerators import (
     PractitionerGenerator,
     Practitioner_QualificationGenerator,
     Practitioner_CommunicationGenerator,
diff --git a/tests/sandbox/generators/test_procedure_generators.py b/tests/sandbox/generators/test_procedure_generators.py
@@ -1,5 +1,5 @@
-from healthchain.data_generators.proceduregenerators import ProcedureGenerator
-from healthchain.data_generators.value_sets.procedurecodes import (
+from healthchain.sandbox.generators.proceduregenerators import ProcedureGenerator
+from healthchain.sandbox.generators.value_sets.procedurecodes import (
     ProcedureCodeSimple,
     ProcedureCodeComplex,
 )
