Updated directory and imports and revised parameter lists.

Signed-off-by: Grace Sng <[email protected]>

Author: grace-sng7

README.md

@@ -43,7 +43,7 @@ factors_list = ["a", "b", "c"]
 treatment = "treatment"
 outcome = "outcome"
 
-#Suggest a list of domain expertises
+# Suggest a list of domain expertises
 domain_expertises = modeler.suggest_domain_expertises(factors_list)
 
 # Suggest a set of potential confounders

pywhyllm/suggesters/identification_suggester.py

@@ -118,7 +118,7 @@ def suggest_backdoor(
             outcome: str,
             factors_list: list(),
             expertise_list: list(),
-            analysis_context=CONTEXT,
+            analysis_context: str = CONTEXT,
             stakeholders: list() = None
     ):
         backdoor_set = self.model_suggester.suggest_confounders(
@@ -138,7 +138,7 @@ def suggest_frontdoor(
             outcome: str,
             factors_list: list(),
             expertise_list: list(),
-            analysis_context=CONTEXT,
+            analysis_context: str = CONTEXT,
             stakeholders: list() = None
     ):
         pass
@@ -149,7 +149,7 @@ def suggest_mediators(
             outcome: str,
             factors_list: list(),
             expertise_list: list(),
-            analysis_context=CONTEXT,
+            analysis_context: str = CONTEXT,
             stakeholders: list() = None
     ):
         expert_list: List[str] = list()
@@ -189,7 +189,7 @@ def request_mediators(
             domain_expertise,
             factors_list,
             mediators_edges,
-            analysis_context=CONTEXT
+            analysis_context: str = CONTEXT
     ):
         mediators: List[str] = list()
 
@@ -262,7 +262,7 @@ def suggest_ivs(
             outcome: str,
             factors_list: list(),
             expertise_list: list(),
-            analysis_context=CONTEXT,
+            analysis_context: str = CONTEXT,
             stakeholders: list() = None
     ):
         expert_list: List[str] = list()

pywhyllm/suggesters/model_suggester.py

@@ -19,7 +19,7 @@ def suggest_domain_expertises(
             self,
             factors_list,
             n_experts: int = 1,
-            analysis_context=CONTEXT
+            analysis_context: str = CONTEXT
     ):
 
         expertise_list: List[str] = list()
@@ -62,7 +62,7 @@ def suggest_domain_experts(
             self,
             factors_list,
             n_experts: int = 5,
-            analysis_context=CONTEXT
+            analysis_context: str = CONTEXT
     ):
 
         experts_list: Set[str] = set()
@@ -104,7 +104,7 @@ def suggest_stakeholders(
             self,
             factors_list,
             n_stakeholders: int = 5,  # must be > 1
-            analysis_context=CONTEXT
+            analysis_context: str = CONTEXT
     ):
 
         stakeholder_list: List[str] = list()
@@ -194,7 +194,7 @@ def request_confounders(
             domain_expertise,
             factors_list,
             confounders_edges,
-            analysis_context=CONTEXT
+            analysis_context: str = CONTEXT
     ):
         confounders: List[str] = list()
 
@@ -265,7 +265,7 @@ def suggest_parents(
             domain_expertise,
             factor,
             factors_list,
-            analysis_context=CONTEXT
+            analysis_context: str = CONTEXT
     ):
         parent_candidates: List[str] = []
 
@@ -330,7 +330,7 @@ def suggest_children(
             domain_expertise,
             factor,
             factors_list,
-            analysis_context=CONTEXT
+            analysis_context: str = CONTEXT
     ):
 
         children_candidates: List[str] = []

pywhyllm/suggesters/validation_suggester.py

@@ -24,7 +24,7 @@ def suggest_negative_controls(
             outcome: str,
             factors_list: list(),
             expertise_list: list(),
-            analysis_context=CONTEXT,
+            analysis_context: str = CONTEXT,
             stakeholders: list() = None
     ):
         expert_list: List[str] = list()
@@ -65,7 +65,7 @@ def request_negative_controls(
             factors_list: list(),
             negative_controls_counter: list(),
             domain_expertise: str,
-            analysis_context=CONTEXT
+            analysis_context: str = CONTEXT
     ):
         negative_controls_list: List[str] = list()
 
@@ -132,7 +132,7 @@ def suggest_latent_confounders(
             treatment: str,
             outcome: str,
             expertise_list: list(),
-            analysis_context=CONTEXT,
+            analysis_context: str = CONTEXT,
             stakeholders: list() = None
     ):
         expert_list: List[str] = list()
@@ -166,7 +166,7 @@ def request_latent_confounders(
             outcome: str,
             latent_confounders_counter: list(),
             domain_expertise: str,
-            analysis_context=CONTEXT
+            analysis_context: str = CONTEXT
     ):
         latent_confounders_list: List[str] = list()
 
@@ -224,7 +224,7 @@ def request_parent_critique(
             factor,
             factors_list,
             domain_expertise,
-            analysis_context=CONTEXT
+            analysis_context: str = CONTEXT
     ):
         edited_factors_list: List[str] = []
 
@@ -275,7 +275,7 @@ def request_children_critique(
             factor,
             factors_list,
             domain_expertise,
-            analysis_context=CONTEXT
+            analysis_context: str = CONTEXT
     ):
         edited_factors_list: List[str] = []