Merge pull request #22 from VACLab/issue-4-gender-prob-bugfix

Bug fix plus enhancement

Author: hyi

biasanalyzer/database.py

@@ -51,6 +51,9 @@ def _initialize(self, db_url, omop_db_url=None):
             else:
                 raise ValueError("Unsupported OMOP database backend")
 
+            # Set self.schema as default schema
+            self.conn.execute(f"SET schema '{self.schema}'")
+
         self._create_cohort_definition_table()
         self._create_cohort_table()
 

biasanalyzer/sql.py

@@ -39,7 +39,7 @@
     SELECT 
         age_bin,
         bin_count,
-        ROUND(bin_count * 1.0 / SUM(bin_count) OVER (), 4) AS probability -- Normalize to get probability
+        ROUND(bin_count * 1.0 / SUM(bin_count) OVER (), 2) AS probability -- Normalize to get probability
     FROM Age_Distribution
     ORDER BY age_bin                  
 '''
@@ -73,7 +73,7 @@
     SELECT 
         gender,
         COALESCE(gender_count, 0) AS gender_count,  -- Ensure that NULL gender counts are treated as 0
-        ROUND(COALESCE(gender_count, 0) * 100.0 / SUM(COALESCE(gender_count, 0)) OVER (), 2) AS probability
+        ROUND(COALESCE(gender_count, 0) * 1.0 / SUM(COALESCE(gender_count, 0)) OVER (), 2) AS probability
     FROM Gender_Distribution
     ORDER BY gender;
 '''