Apply formatting

nikhilwoodruff · Jul 19, 2021 · 084d35c · 084d35c
1 parent c8adbd7
commit 084d35c
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/openfisca_uk_data/datasets/__init__.py b/openfisca_uk_data/datasets/__init__.py
@@ -3,6 +3,6 @@
     FRS,
     BaseFRS,
     FRS_SPI_Adjusted,
-    SynthFRS
+    SynthFRS,
 )
 from openfisca_uk_data.datasets.spi import RawSPI, BaseSPI, SPI
diff --git a/openfisca_uk_data/datasets/frs/__init__.py b/openfisca_uk_data/datasets/frs/__init__.py
@@ -3,4 +3,4 @@
 from openfisca_uk_data.datasets.frs.base_frs import BaseFRS
 from openfisca_uk_data.datasets.frs.frs import FRS
 from openfisca_uk_data.datasets.frs.frs_spi_adjusted import FRS_SPI_Adjusted
-from openfisca_uk_data.datasets.frs.synth_frs import SynthFRS
+from openfisca_uk_data.datasets.frs.synth_frs import SynthFRS
diff --git a/openfisca_uk_data/datasets/frs/synth_frs.py b/openfisca_uk_data/datasets/frs/synth_frs.py
@@ -5,6 +5,7 @@
 import numpy as np
 import h5py
 
+
 @dataset
 class SynthFRS:
     name = "synth_frs"
@@ -47,7 +48,9 @@ def anonymise(df: pd.DataFrame) -> pd.DataFrame:
             for variable in data.keys():
                 if system.variables[variable].entity.key == entity:
                     entity_dfs[entity][variable] = data[variable][str(year)]
-        person, benunit, household = map(lambda x: anonymise(pd.DataFrame(x)), entity_dfs.values())
+        person, benunit, household = map(
+            lambda x: anonymise(pd.DataFrame(x)), entity_dfs.values()
+        )
 
         year = int(year)
 
@@ -61,6 +64,5 @@ def anonymise(df: pd.DataFrame) -> pd.DataFrame:
             for variable in household.columns:
                 f[f"{variable}/{year}"] = household[variable].values
 
-
     def save(data_file, year: int = 2018):
-        shutil.copyfile(data_file, SynthFRS.data_dir / SynthFRS.filename(year))
+        shutil.copyfile(data_file, SynthFRS.data_dir / SynthFRS.filename(year))