feat: making static analysis as the default option for the TEE (#274)

limebit · Jan 15, 2025 · 816ee6f · 816ee6f
1 parent 75253a4
commit 816ee6f
Show file tree

Hide file tree

Showing 4 changed files with 525 additions and 261 deletions.
diff --git a/medmodels/treatment_effect/builder.py b/medmodels/treatment_effect/builder.py
@@ -23,6 +23,9 @@ class TreatmentEffectBuilder:
     The TreatmentEffectBuilder class is used to build a TreatmentEffect object with
     the desired configurations for the treatment effect estimation using a builder
     pattern.
+
+    By default, it configures a static treatment effect estimation. To configure a
+    time-dependent treatment effect estimation, the time_attribute must be set.
     """
 
     treatment: Group
@@ -97,6 +100,9 @@ def with_time_attribute(
     ) -> TreatmentEffectBuilder:
         """Sets the time attribute to be used in the treatment effect estimation.
 
+        It turs the treatment effect estimation from a static to a time-dependent
+        analysis.
+
         Args:
             attribute (MedRecordAttribute): The time attribute.