Black reformating

romainsacchi · Mar 17, 2022 · 4f9e71f · 4f9e71f
1 parent 618adab
commit 4f9e71f
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/carculator/hot_emissions.py b/carculator/hot_emissions.py
@@ -5,8 +5,8 @@
 from typing import List, Union
 
 import numpy as np
-import xarray as xr
 import pandas as pd
+import xarray as xr
 
 from . import DATA_DIR
 
@@ -24,7 +24,12 @@ def get_hot_emission_factors() -> xr.DataArray:
     """
     filepath = DATA_DIR / "hot.csv"
 
-    hot = pd.read_csv(filepath).groupby(["variable", "powertrain", "euro_class", "component"])["hot"].mean().to_xarray()
+    hot = (
+        pd.read_csv(filepath)
+        .groupby(["variable", "powertrain", "euro_class", "component"])["hot"]
+        .mean()
+        .to_xarray()
+    )
 
     return hot
 
@@ -35,7 +40,12 @@ def get_non_hot_emission_factors() -> xr.DataArray:
     """
     filepath = DATA_DIR / "non_hot.csv"
 
-    non_hot = pd.read_csv(filepath).groupby(["powertrain", "euro_class", "type", "Component"])["hot"].mean().to_xarray()
+    non_hot = (
+        pd.read_csv(filepath)
+        .groupby(["powertrain", "euro_class", "type", "Component"])["hot"]
+        .mean()
+        .to_xarray()
+    )
 
     return non_hot