Fix auto-merge issue

f-dangel · Nov 7, 2023 · ab33832 · ab33832
1 parent a70f4e0
commit ab33832
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/test_kfac.py b/test/test_kfac.py
@@ -56,6 +56,10 @@ def test_kfac(
 
  report_nonclose(ggn, kfac_mat)
 
+ # Check that input covariances were not computed
+ if exclude == "weight":
+ assert len(kfac._input_covariances) == 0
+
 
 @mark.parametrize("shuffle", [False, True], ids=["", "shuffled"])
 def test_kfac_mc(
@@ -91,10 +95,6 @@ def test_kfac_mc(
 
  report_nonclose(ggn, kfac_mat, rtol=rtol, atol=atol)
 
- # Check that input covariances were not computed
- if exclude == "weight":
- assert len(kfac._input_covariances) == 0
-
 
 def test_kfac_one_datum(
  kfac_expand_exact_one_datum_case: Tuple[