Fix broken tests

mtfishman · mtfishman · commit ca65944f6400 · 2025-02-20T16:32:00.000-05:00
diff --git a/NDTensors/src/blocksparse/contract_generic.jl b/NDTensors/src/blocksparse/contract_generic.jl
@@ -11,8 +11,8 @@ function contract_blockoffsets(
   indsR,
   labelsR,
 )
-  N1 = ndims(boffs1)
-  N2 = ndims(boffs2)
+  N1 = length(blocktype(boffs1))
+  N2 = length(blocktype(boffs2))
   NR = length(labelsR)
   ValNR = ValLength(labelsR)
   labels1_to_labels2, labels1_to_labelsR, labels2_to_labelsR = contract_labels(
diff --git a/NDTensors/src/blocksparse/contract_sequential.jl b/NDTensors/src/blocksparse/contract_sequential.jl
@@ -9,8 +9,8 @@ function contract_blockoffsets(
   indsR,
   labelsR,
 )
-  N1 = length(keytype(boffs1))
-  N2 = length(keytype(boffs2))
+  N1 = length(blocktype(boffs1))
+  N2 = length(blocktype(boffs2))
   NR = length(labelsR)
   ValNR = ValLength(labelsR)
   labels1_to_labels2, labels1_to_labelsR, labels2_to_labelsR = contract_labels(
