formatting

NREL · Nov 17, 2023 · 084031b · 084031b
1 parent 6b04055
commit 084031b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 10 deletions.
diff --git a/nrel/routee/powertrain/validation/errors.py b/nrel/routee/powertrain/validation/errors.py
@@ -28,9 +28,7 @@
 }
 
 
-def mean_squared_error(
-    A, B, axis: Optional[int] = None
-) -> float:
+def mean_squared_error(A, B, axis: Optional[int] = None) -> float:
     return np.square(A - B).mean(axis=axis)
 
 
@@ -41,9 +39,7 @@ def net_energy_error(target, target_pred) -> float:
     return net_error
 
 
-def weighted_relative_percent_difference(
-    target, target_pred
-) -> float:
+def weighted_relative_percent_difference(target, target_pred) -> float:
     epsilon = np.finfo(np.float64).eps
 
     w = np.array(np.abs(target) / np.sum(np.abs(target)))

diff --git a/scripts/developers/train_model_catalog.py b/scripts/developers/train_model_catalog.py
@@ -312,25 +312,25 @@ def load_all_files(files, file_limit=FILE_LIMIT):
     pt.DataColumn(name="grade_dec", units="decimal"),
     pt.DataColumn(name="entry_angle", units="degrees"),
 ]
-features = [ 
+features = [
     [pt.DataColumn(name="speed_mph", units="mph")],
     [
         pt.DataColumn(name="speed_mph", units="mph"),
-        pt.DataColumn(name="grade_dec", units="decimal")
+        pt.DataColumn(name="grade_dec", units="decimal"),
     ],
     [
         pt.DataColumn(name="speed_mph", units="mph"),
         pt.DataColumn(name="previous_speed_mph", units="mph"),
         pt.DataColumn(name="grade_dec", units="decimal"),
-        pt.DataColumn(name="previous_grade_dec", units="decimal")
+        pt.DataColumn(name="previous_grade_dec", units="decimal"),
     ],
     [
         pt.DataColumn(name="previous_speed_mph", units="mph"),
         pt.DataColumn(name="speed_mph", units="mph"),
         pt.DataColumn(name="previous_grade_dec", units="decimal"),
         pt.DataColumn(name="grade_dec", units="decimal"),
         pt.DataColumn(name="entry_angle", units="degrees"),
-        pt.DataColumn(name="exit_angle", units="degrees")
+        pt.DataColumn(name="exit_angle", units="degrees"),
     ],
 ]