chore: fix format

WIP fixes: PolicyEngine#274
SylviaDu99 · Oct 16, 2024 · 56582af · 56582af
1 parent d9a37d5
commit 56582af
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 10 deletions.
diff --git a/policyengine_core/parameters/parameter_node.py b/policyengine_core/parameters/parameter_node.py
@@ -284,7 +284,9 @@ def write_yaml(self, file_path: Path) -> yaml:
             value_dict = {}
             data[name] = {"values": value_dict}
             for value_at_instant in value.values_list:
-                value_dict[value_at_instant.instant_str] = value_at_instant.value
+                value_dict[value_at_instant.instant_str] = (
+                    value_at_instant.value
+                )
         try:
             with open(file_path, "w") as f:
                 yaml.dump(data, f, sort_keys=False)

diff --git a/tests/core/test_parameters.py b/tests/core/test_parameters.py
@@ -145,17 +145,17 @@ def test_name():
 
 
 def test_write_yaml():
-    parameter_data={
-        'amount': {
-            'values': {
-                "2015-01-01": {'value': 550},
-                "2016-01-01": {'value': 600}
+    parameter_data = {
+        "amount": {
+            "values": {
+                "2015-01-01": {"value": 550},
+                "2016-01-01": {"value": 600},
             }
         },
-        'min_age': {
-            'values': {
-                "2015-01-01": {'value': 25},
-                "2016-01-01": {'value': 18}
+        "min_age": {
+            "values": {
+                "2015-01-01": {"value": 25},
+                "2016-01-01": {"value": 18},
             }
         },
     }