v2 cutover

smart-on-fhir · Apr 3, 2024 · 1ffb869 · 1ffb869
1 parent 368507d
commit 1ffb869
Show file tree

Hide file tree

Showing 7 changed files with 949 additions and 297 deletions.
diff --git a/cumulus_library_covid/covid_symptom/counts.py b/cumulus_library_covid/covid_symptom/counts.py
@@ -1,5 +1,5 @@
 from pathlib import Path
-from cumulus_library.schema.counts import CountsBuilder
+from cumulus_library.statistics.counts import CountsBuilder
 
 
 class CovidCountsBuilder(CountsBuilder):
@@ -98,7 +98,7 @@ def count_symptom(self, duration="week"):
         ]
         return self.count_encounter(view_name, from_table, cols)
 
-    def prepare_queries(self, cursor=None, schema=None):
+    def prepare_queries(self, cursor=None, schema=None, **kwargs):
         self.queries = [
             self.count_dx("month"),
             self.count_dx("week"),
@@ -112,7 +112,6 @@ def prepare_queries(self, cursor=None, schema=None):
             self.count_prevalence_ed("week"),
         ]
 
-
 if __name__ == "__main__":
     builder = CovidCountsBuilder()
-    builder.write_counts(f"{Path(__file__).resolve().parent}/counts.sql")
+    builder.write_counts(f"{Path(__file__).resolve().parent}/counts.sql")