fix bug

logix-project · Apr 22, 2024 · dde59e3 · dde59e3
1 parent d9b5a1a
commit dde59e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/logix/analysis/influence_function_utils.py b/logix/analysis/influence_function_utils.py
@@ -50,7 +50,7 @@ def precondition_kfac(
             "a b, batch b c, c d -> batch a d",
         )
 
-        return preconditioned
+    return preconditioned
 
 
 def precondition_raw(