Lint fix

pomonam · Jul 11, 2024 · 7220ecc · 7220ecc
1 parent d7e35bc
commit 7220ecc
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/examples/cifar/inspect_factors.py b/examples/cifar/inspect_factors.py
@@ -9,16 +9,12 @@ def main():
     logging.basicConfig(level=logging.INFO)
 
     name = "ekfac"
-    factor = (
-        Analyzer.load_file(f"influence_results/cifar10/factors_{name}/activation_covariance.safetensors")
-    )
+    factor = Analyzer.load_file(f"influence_results/cifar10/factors_{name}/activation_covariance.safetensors")
 
     plt.matshow(factor["6.0"])
     plt.show()
 
-    factor = (
-        Analyzer.load_file(f"influence_results/cifar10/factors_{name}/gradient_covariance.safetensors")
-    )
+    factor = Analyzer.load_file(f"influence_results/cifar10/factors_{name}/gradient_covariance.safetensors")
 
     plt.matshow(factor["6.0"])
     plt.show()