update

sdaza · Dec 12, 2024 · dec3a36 · dec3a36
1 parent 9edc7c3
commit dec3a36
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/experiment_utils/experiment_analyzer.py b/experiment_utils/experiment_analyzer.py
@@ -581,13 +581,12 @@ def get_effects(self, min_binary_count=100, adjustment=None):
             relevant_covariates = set(self.final_covariates) & set(self.regression_covariates)
 
             adjustment_labels = {
-                None: '',
                 'IPW': 'IPW',
                 'IV': 'IV'
             }
 
             if adjustment in adjustment_labels and len(relevant_covariates) > 0:
-                adjustment_label = adjustment_labels[adjustment] + ' + Regression'
+                adjustment_label = adjustment_labels[adjustment] + '+Regression'
             elif adjustment in adjustment_labels:
                 adjustment_label = adjustment_labels[adjustment]
             elif len(relevant_covariates) > 0: