make tests pass + formatting

plutonium-239 · Aug 1, 2024 · 49d0f92 · 49d0f92
1 parent d7d6c3b
commit 49d0f92
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/experiments/visual_abstract/plot_data.py b/experiments/visual_abstract/plot_data.py
@@ -97,6 +97,9 @@
 
             plt.legend()
             plt.savefig(
-                path.join(HEREDIR, f"visual_abstract_{architecture}_{mode}{'_use_compile' if use_compile else ''}.pdf"),
+                path.join(
+                    HEREDIR,
+                    f"visual_abstract_{architecture}_{mode}{'_use_compile' if use_compile else ''}.pdf",
+                ),
                 bbox_inches="tight",
             )
diff --git a/memsave_torch/nn/functional/Conv.py b/memsave_torch/nn/functional/Conv.py
@@ -86,7 +86,7 @@ def backward(ctx, grad_output):
             ctx.needs_input_grad[:3],
         )
 
-        return grad_x, grad_weight, grad_bias, None, None, None, None, None
+        return grad_x, grad_weight, grad_bias, None, None, None, None, None, None
 
 
 def convMemSave(