Format

PolicyEngine · Oct 13, 2022 · abc1e43 · abc1e43
1 parent 5291104
commit abc1e43
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 8 deletions.
diff --git a/docs/usage/datasets.ipynb b/docs/usage/datasets.ipynb
@@ -52,10 +52,9 @@
     "        }\n",
     "        self.save_variable_values(year, data)\n",
     "\n",
+    "\n",
     "# Important: we must instantiate datasets. This tests their validity and adds dynamic logic.\n",
-    "CountryTemplateDataset = (\n",
-    "    CountryTemplateDataset()\n",
-    ")"
+    "CountryTemplateDataset = CountryTemplateDataset()"
    ]
   },
   {

diff --git a/docs/usage/simulation.ipynb b/docs/usage/simulation.ipynb
@@ -77,7 +77,7 @@
     "\n",
     "sim = Microsimulation()\n",
     "\n",
-    "sim.calculate(\"income_tax\", \"2022-01\").sum()/1e6"
+    "sim.calculate(\"income_tax\", \"2022-01\").sum() / 1e6"
    ]
   },
   {

diff --git a/policyengine_core/country_template/data/datasets/country_template_dataset.py b/policyengine_core/country_template/data/datasets/country_template_dataset.py
@@ -31,7 +31,6 @@ def generate(self, year: int) -> None:
         }
         self.save_variable_values(year, data)
 
+
 # Important: we must instantiate datasets. This tests their validity and adds dynamic logic.
-CountryTemplateDataset = (
-    CountryTemplateDataset()
-)
+CountryTemplateDataset = CountryTemplateDataset()
diff --git a/setup.py b/setup.py
@@ -11,7 +11,7 @@
 general_requirements = [
     "pytest>=4,<6",
     "numpy>=1.11,<1.21",
-    "black<23",
+    "black[jupyter]<23",
     "linecheck<1",
     "yaml-changelog<1",
     "coverage<7",
-Original file line number
+Diff line change
@@ Expand Up / @@ -77,7 +77,7 @@ @@
         "\n",
         "sim = Microsimulation()\n",
         "\n",
-        "sim.calculate(\"income_tax\", \"2022-01\").sum()/1e6"
+        "sim.calculate(\"income_tax\", \"2022-01\").sum() / 1e6"
        ]
       },
       {
@@ Expand Down @@