From 2a2b22fb95a761bdaabd0ecb0a8e2c320330b6cb Mon Sep 17 00:00:00 2001
From: Casey Jao <casey@agnostiq.ai>
Date: Tue, 28 May 2024 12:51:54 -0400
Subject: [PATCH] Fix pre-commit

---
 covalent/_dispatcher_plugins/local.py                     | 4 ++--
 .../_db/write_result_to_db_test.py                        | 8 ++++----
 tests/stress_tests/scripts/mnist_sublattices.py           | 2 +-
 tests/stress_tests/scripts/sublattices_mixed.py           | 2 +-
 tests/stress_tests/scripts/tasks.py                       | 2 +-
 5 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/covalent/_dispatcher_plugins/local.py b/covalent/_dispatcher_plugins/local.py
index 086b8dec2..9857342cf 100644
--- a/covalent/_dispatcher_plugins/local.py
+++ b/covalent/_dispatcher_plugins/local.py
@@ -600,7 +600,7 @@ def _upload(assets: List[AssetSchema]):
         number_uploaded = 0
         for i, asset in enumerate(assets):
             if not asset.remote_uri or not asset.uri:
-                app_log.debug(f"Skipping asset {i+1} out of {total}")
+                app_log.debug(f"Skipping asset {i + 1} out of {total}")
                 continue
             if asset.remote_uri.startswith(local_scheme_prefix):
                 copy_file_locally(asset.uri, asset.remote_uri)
@@ -608,7 +608,7 @@ def _upload(assets: List[AssetSchema]):
             else:
                 _upload_asset(asset.uri, asset.remote_uri)
                 number_uploaded += 1
-            app_log.debug(f"Uploaded asset {i+1} out of {total}.")
+            app_log.debug(f"Uploaded asset {i + 1} out of {total}.")
         app_log.debug(f"uploaded {number_uploaded} assets.")
 
 
diff --git a/tests/covalent_dispatcher_tests/_db/write_result_to_db_test.py b/tests/covalent_dispatcher_tests/_db/write_result_to_db_test.py
index b383dd253..310367df7 100644
--- a/tests/covalent_dispatcher_tests/_db/write_result_to_db_test.py
+++ b/tests/covalent_dispatcher_tests/_db/write_result_to_db_test.py
@@ -245,8 +245,8 @@ def test_insert_lattices_data(test_db, mocker):
         lattice_args = get_lattice_kwargs(
             dispatch_id=f"dispatch_{i + 1}",
             name=f"workflow_{i + 1}",
-            docstring_filename=f"docstring_{i+1}.txt",
-            storage_path=f"results/dispatch_{i+1}/",
+            docstring_filename=f"docstring_{i + 1}.txt",
+            storage_path=f"results/dispatch_{i + 1}/",
             executor="dask",
             workflow_executor="dask",
             created_at=cur_time,
@@ -264,10 +264,10 @@ def test_insert_lattices_data(test_db, mocker):
             assert lattice.dispatch_id == f"dispatch_{i + 1}"
             assert lattice.electron_id is None
             assert lattice.name == f"workflow_{i + 1}"
-            assert lattice.docstring_filename == f"docstring_{i+1}.txt"
+            assert lattice.docstring_filename == f"docstring_{i + 1}.txt"
             assert lattice.status == "RUNNING"
             assert lattice.storage_type == STORAGE_TYPE
-            assert lattice.storage_path == f"results/dispatch_{i+1}/"
+            assert lattice.storage_path == f"results/dispatch_{i + 1}/"
             assert lattice.function_filename == FUNCTION_FILENAME
             assert lattice.function_string_filename == FUNCTION_STRING_FILENAME
             assert lattice.executor == "dask"
diff --git a/tests/stress_tests/scripts/mnist_sublattices.py b/tests/stress_tests/scripts/mnist_sublattices.py
index 31ad46fc8..2b0e0aab0 100644
--- a/tests/stress_tests/scripts/mnist_sublattices.py
+++ b/tests/stress_tests/scripts/mnist_sublattices.py
@@ -146,7 +146,7 @@ def test(
             correct += (pred.argmax(1) == y).type(torch.float).sum().item()
     test_loss /= num_batches
     correct /= size
-    print(f"Test Error: \n Accuracy: {(100*correct):>0.1f}%, Avg loss: {test_loss:>8f} \n")
+    print(f"Test Error: \n Accuracy: {(100 * correct):>0.1f}%, Avg loss: {test_loss:>8f} \n")
 
 
 def train_model(
diff --git a/tests/stress_tests/scripts/sublattices_mixed.py b/tests/stress_tests/scripts/sublattices_mixed.py
index 4dc085f0f..fda5f8c27 100644
--- a/tests/stress_tests/scripts/sublattices_mixed.py
+++ b/tests/stress_tests/scripts/sublattices_mixed.py
@@ -147,7 +147,7 @@ def test(
             correct += (pred.argmax(1) == y).type(torch.float).sum().item()
     test_loss /= num_batches
     correct /= size
-    print(f"Test Error: \n Accuracy: {(100*correct):>0.1f}%, Avg loss: {test_loss:>8f} \n")
+    print(f"Test Error: \n Accuracy: {(100 * correct): > 0.1f}%, Avg loss: {test_loss: > 8f} \n")
 
 
 def train_model(
diff --git a/tests/stress_tests/scripts/tasks.py b/tests/stress_tests/scripts/tasks.py
index 55d9ab8c9..181c4c33a 100644
--- a/tests/stress_tests/scripts/tasks.py
+++ b/tests/stress_tests/scripts/tasks.py
@@ -175,7 +175,7 @@ def test(
             correct += (pred.argmax(1) == y).type(torch.float).sum().item()
     test_loss /= num_batches
     correct /= size
-    print(f"Test Error: \n Accuracy: {(100*correct):>0.1f}%, Avg loss: {test_loss:>8f} \n")
+    print(f"Test Error: \n Accuracy: {(100 * correct):>0.1f}%, Avg loss: {test_loss:>8f} \n")
 
 
 def train_model(