Format

Author: mtfishman

NDTensors/src/BlockSparseArrays/src/blocksparsearray.jl

@@ -17,21 +17,29 @@ struct BlockZero{Axes}
   axes::Axes
 end
 
-(f::BlockZero)(T::Type, I::CartesianIndex) = fill!(T(undef, block_size(f.axes, Block(Tuple(I)))), false)
+function (f::BlockZero)(T::Type, I::CartesianIndex)
+  return fill!(T(undef, block_size(f.axes, Block(Tuple(I)))), false)
+end
 
-function BlockSparseArray(blocks::AbstractVector{<:Block{N}}, blockdata::AbstractVector, axes::NTuple{N}) where {N}
+function BlockSparseArray(
+  blocks::AbstractVector{<:Block{N}}, blockdata::AbstractVector, axes::NTuple{N}
+) where {N}
   return BlockSparseArray(Dictionary(blocks, blockdata), axes)
 end
 
-function BlockSparseArray(blockdata::Dictionary{<:Block{N}}, axes::NTuple{N,AbstractUnitRange{Int}}) where {N}
+function BlockSparseArray(
+  blockdata::Dictionary{<:Block{N}}, axes::NTuple{N,AbstractUnitRange{Int}}
+) where {N}
   blocks = keys(blockdata)
   cartesianblocks = map(block -> CartesianIndex(block.n), blocks)
   cartesiandata = Dictionary(cartesianblocks, blockdata)
   block_storage = SparseArray(cartesiandata, blocklength.(axes), BlockZero(axes))
   return BlockSparseArray(block_storage, axes)
 end
 
-function BlockSparseArray(blockdata::Dictionary{<:Block{N}}, blockinds::NTuple{N,AbstractVector}) where {N}
+function BlockSparseArray(
+  blockdata::Dictionary{<:Block{N}}, blockinds::NTuple{N,AbstractVector}
+) where {N}
   return BlockSparseArray(blockdata, blockedrange.(blockinds))
 end
 

NDTensors/src/BlockSparseArrays/test/runtests.jl

@@ -3,7 +3,10 @@ using NDTensors.BlockSparseArrays
 
 @testset "Test NDTensors.BlockSparseArrays" begin
   @testset "README" begin
-    @test include(joinpath(pkgdir(BlockSparseArrays), "src", "BlockSparseArrays", "examples", "README.jl")) isa Any
+    @test include(
+      joinpath(
+        pkgdir(BlockSparseArrays), "src", "BlockSparseArrays", "examples", "README.jl"
+      ),
+    ) isa Any
   end
 end
-

NDTensors/src/arraytensor/arraytensor.jl

@@ -1,7 +1,12 @@
 # Used for dispatch to distinguish from Tensors wrapping TensorStorage.
 # Remove once TensorStorage is removed.
 const ArrayStorage{T,N} = Union{
-  Array{T,N},ReshapedArray{T,N},SubArray{T,N},PermutedDimsArray{T,N},StridedView{T,N},BlockSparseArray{T,N}
+  Array{T,N},
+  ReshapedArray{T,N},
+  SubArray{T,N},
+  PermutedDimsArray{T,N},
+  StridedView{T,N},
+  BlockSparseArray{T,N},
 }
 const MatrixStorage{T} = Union{
   ArrayStorage{T,2},

NDTensors/src/arraytensor/blocksparsearray.jl

@@ -1,7 +1,5 @@
 # TODO: Implement.
-function contraction_output(
-  tensor1::BlockSparseArray, tensor2::BlockSparseArray, indsR
-)
+function contraction_output(tensor1::BlockSparseArray, tensor2::BlockSparseArray, indsR)
   return error("Not implemented")
 end