From 48614e3c2b9a6c469da300ac7ec47a2c401bbbb5 Mon Sep 17 00:00:00 2001
From: kcreekdev <158532564+kcreekdev@users.noreply.github.com>
Date: Sun, 21 Apr 2024 17:14:58 -0400
Subject: [PATCH] Fix format errors

---
 cs-config/cs_config/functions.py | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/cs-config/cs_config/functions.py b/cs-config/cs_config/functions.py
index 206d1ae1..4e41fb48 100644
--- a/cs-config/cs_config/functions.py
+++ b/cs-config/cs_config/functions.py
@@ -211,8 +211,12 @@ def run_model(meta_param_dict, adjustment):
 
     # Dask parmeters
     num_workers = 2
-    memory_per_worker = "10GiB"
-    client = Client(n_workers=num_workers, threads_per_worker=1, memory_limit=memory_per_worker)
+    memory_limit = "10GiB"
+    client = Client(
+        n_workers=num_workers,
+        threads_per_worker=1,
+        memory_limit=memory_per_worker,
+    )
     # TODO: Swap to these parameters when able to specify tax function
     # and model workers separately
     # num_workers_txf = 5
@@ -294,7 +298,11 @@ def run_model(meta_param_dict, adjustment):
     )
     client.close()
     del client
-    client = Client(n_workers=num_workers, threads_per_worker=1, memory_limit=memory_per_worker)
+    client = Client(
+        n_workers=num_workers,
+        threads_per_worker=1,
+        memory_limit=memory_per_worker,
+    )
     # update tax function parameters in Specifications Object
     d_base = c_base.get_dict()
     # additional parameters to change
@@ -311,7 +319,11 @@ def run_model(meta_param_dict, adjustment):
     base_ss_dir = os.path.join(base_dir, "SS", "SS_vars.pkl")
     client.close()
     del client
-    client = Client(n_workers=num_workers, threads_per_worker=1, memory_limit=memory_per_worker)
+    client = Client(
+        n_workers=num_workers,
+        threads_per_worker=1,
+        memory_limit=memory_per_worker,
+    )
     with open(base_ss_dir, "wb") as f:
         pickle.dump(base_ss, f)
     if time_path: