black

cumulus_library_covid/covid_symptom/count.py

@@ -1,6 +1,7 @@
 from pathlib import Path
 from cumulus_library.schema.counts import CountsBuilder
 
+
 class CovidCountsBuilder(CountsBuilder):
     display_text = "Creating covid counts..."
 
@@ -20,7 +21,6 @@ def count_dx(self, duration="week"):
         ]
         return self.count_encounter(view_name, from_table, cols)
 
-
     def count_pcr(self, duration="week"):
         """
         covid_symptom__count_pcr_week
@@ -39,7 +39,6 @@ def count_pcr(self, duration="week"):
         ]
         return self.count_encounter(view_name, from_table, cols)
 
-
     def count_study_period(self, duration="month"):
         """
         covid_symptom__count_study_period_week
@@ -58,7 +57,6 @@ def count_study_period(self, duration="month"):
         ]
         return self.count_encounter(view_name, from_table, cols)
 
-
     def count_prevalence_ed(self, duration="month"):
         view_name = self.get_table_name("count_prevalence_ed", duration)
         from_table = self.get_table_name("prevalence_ed")
@@ -75,7 +73,6 @@ def count_prevalence_ed(self, duration="month"):
         ]
         return self.count_encounter(view_name, from_table, cols)
 
-
     def count_symptom(self, duration="week"):
         """
         covid_symptom__count_symptom_week
@@ -95,7 +92,6 @@ def count_symptom(self, duration="week"):
         ]
         return self.count_encounter(view_name, from_table, cols)
 
-
     def prepare_queries(self, cursor=None, schema=None):
         self.queries = [
             self.count_dx("month"),
@@ -110,6 +106,7 @@ 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}/count.sql")
+    builder.write_counts(f"{Path(__file__).resolve().parent}/count.sql")