format yas style

Author: Jutho

.JuliaFormatter.toml

@@ -0,0 +1 @@
+style = "yas"

docs/make.jl

@@ -1,19 +1,17 @@
 using Documenter
 using TensorOperations
 
-makedocs(modules=[TensorOperations],
-            sitename = "TensorOperations.jl",
-            authors = "Jutho Haegeman",
-            format = Documenter.HTML(prettyurls = get(ENV, "CI", nothing) == "true"),
-            pages = [
-                "Home" => ["index.md",
-                            "indexnotation.md",
-                            "functions.md",
-                            "cache.md",
-                            "implementation.md"]
-            ])
+makedocs(; modules=[TensorOperations],
+         sitename="TensorOperations.jl",
+         authors="Jutho Haegeman",
+         format=Documenter.HTML(; prettyurls=get(ENV, "CI", nothing) == "true"),
+         pages=["Home" => ["index.md",
+                           "indexnotation.md",
+                           "functions.md",
+                           "cache.md",
+                           "implementation.md"]])
 
 # Documenter can also automatically deploy documentation to gh-pages.
 # See "Hosting Documentation" and deploydocs() in the Documenter manual
 # for more information.
-deploydocs(repo = "github.com/Jutho/TensorOperations.jl.git")
+deploydocs(; repo="github.com/Jutho/TensorOperations.jl.git")

src/TensorOperations.jl

@@ -4,10 +4,13 @@
 tensorcopy!(A, IA, C, IC) = tensorcopy!(A, tuple(IA...), C, tuple(IC...))
 tensoradd!(α, A, IA, β, C, IC) = tensoradd!(α, A, tuple(IA...), β, C, tuple(IC...))
 tensortrace!(α, A, IA, β, C, IC) = tensortrace!(α, A, tuple(IA...), β, C, tuple(IC...))
-tensorcontract!(α, A, IA, conjA, B, IB, conjB, β, C, IC) =
-    tensorcontract!(α, A, tuple(IA...), conjA, B, tuple(IB...), conjB, β, C, tuple(IC...))
-tensorproduct!(α, A, IA, B, IB, β, C, IC) = tensorproduct!(α, A, tuple(IA...), B, tuple(IB...), β, C, tuple(IC...))
-
+function tensorcontract!(α, A, IA, conjA, B, IB, conjB, β, C, IC)
+    return tensorcontract!(α, A, tuple(IA...), conjA, B, tuple(IB...), conjB, β, C,
+                           tuple(IC...))
+end
+function tensorproduct!(α, A, IA, B, IB, β, C, IC)
+    return tensorproduct!(α, A, tuple(IA...), B, tuple(IB...), β, C, tuple(IC...))
+end
 
 """
     tensorcopy!(A, IA, C, IC)
@@ -30,7 +33,7 @@ See also: [`tensorcopy`](@ref)
 """
 function tensoradd!(α, A, IA::Tuple, β, C, IC::Tuple)
     indCinA = add_indices(IA, IC)
-    add!(α, A, :N, β, C, indCinA)
+    return add!(α, A, :N, β, C, indCinA)
 end
 
 """
@@ -60,9 +63,10 @@ or in the interaction of either `labelsA` or `labelsB` with `labelsC`, for indic
 in which the open indices should be match to the indices of the output array `C`.
 """
 function tensorcontract!(α, A, IA::Tuple, conjA, B, IB::Tuple, conjB, β, C, IC::Tuple)
-
-    conjA == 'N' || conjA == 'C' || throw(ArgumentError("Value of conjA should be 'N' or 'C' instead of $conjA"))
-    conjB == 'N' || conjB == 'C' || throw(ArgumentError("Value of conjB should be 'N' or 'C' instead of $conjB"))
+    conjA == 'N' || conjA == 'C' ||
+        throw(ArgumentError("Value of conjA should be 'N' or 'C' instead of $conjA"))
+    conjB == 'N' || conjB == 'C' ||
+        throw(ArgumentError("Value of conjB should be 'N' or 'C' instead of $conjB"))
     CA = conjA == 'N' ? :N : :C
     CB = conjB == 'N' ? :N : :C
 
@@ -78,5 +82,5 @@ Replaces C with `β C + α A * B` without any indices being contracted.
 """
 function tensorproduct!(α, A, IA::Tuple, B, IB::Tuple, β, C, IC::Tuple)
     isempty(intersect(IA, IB)) || throw(LabelError("not a valid tensor product"))
-    tensorcontract!(α, A, IA, 'N', B, IB, 'N', β, C, IC)
+    return tensorcontract!(α, A, IA, 'N', B, IB, 'N', β, C, IC)
 end

src/functions/inplace.jl

@@ -25,8 +25,8 @@ over are labelled by increasing integers, i.e. first the contraction correspondi
 See also the macro version [`@ncon`](@ref).
 """
 function ncon(tensors, network,
-                conjlist = fill(false, length(tensors)), sym = nothing ;
-                order = nothing, output = nothing)
+              conjlist=fill(false, length(tensors)), sym=nothing;
+              order=nothing, output=nothing)
     # length(tensors) >= 2 ||
     #     throw(ArgumentError("do not use `ncon` for less than two tensors"))
     length(tensors) == length(network) == length(conjlist) ||
@@ -41,7 +41,7 @@ function ncon(tensors, network,
         end
     end
     if output === nothing
-        output = sort(outputindices; rev = true)
+        output = sort(outputindices; rev=true)
     else
         for a in output
             a in outputindices ||
@@ -59,7 +59,7 @@ function ncon(tensors, network,
             return tensortrace(tensors[1], network[1], output)
         end
     end
-    (tensors,network) = resolve_traces(tensors,network);
+    (tensors, network) = resolve_traces(tensors, network)
     tree = order === nothing ? ncontree(network) : indexordertree(network, order)
 
     if sym !== nothing
@@ -82,7 +82,7 @@ function ncon(tensors, network,
         symcontract = nothing
     end
     contract!(true, A, CA, B, CB, false, C,
-                oindA, cindA, oindB, cindB, indCinoAB, (), symcontract)
+              oindA, cindA, oindB, cindB, indCinoAB, (), symcontract)
     return C
 end
 
@@ -116,6 +116,6 @@ function contracttree(tensors, network, conjlist, tree, sym)
         symcontract = nothing
     end
     contract!(true, A, CA, B, CB, false, C,
-                oindA, cindA, oindB, cindB, indCinoAB, (), symcontract)
+              oindA, cindA, oindB, cindB, indCinoAB, (), symcontract)
     return C, IC, :N
 end

src/functions/ncon.jl

@@ -5,10 +5,13 @@
 # type unstable; better use tuples instead
 tensorcopy(A, IA, IC=IA) = tensorcopy(A, tuple(IA...), tuple(IC...))
 tensoradd(A, IA, B, IB, IC=IA) = tensoradd(A, tuple(IA...), B, tuple(IB...), tuple(IC...))
-tensortrace(A, IA, IC = unique2(IA)) = tensortrace(A, tuple(IA...), tuple(IC...))
-tensorcontract(A, IA, B, IB, IC = symdiff(IA, IB)) =
-    tensorcontract(A, tuple(IA...), B, tuple(IB...), tuple(IC...))
-tensorproduct(A, IA, B, IB, IC = vcat(IA, IB)) = tensorproduct(A, tuple(IA...), B, tuple(IB...), tuple(IC...))
+tensortrace(A, IA, IC=unique2(IA)) = tensortrace(A, tuple(IA...), tuple(IC...))
+function tensorcontract(A, IA, B, IB, IC=symdiff(IA, IB))
+    return tensorcontract(A, tuple(IA...), B, tuple(IB...), tuple(IC...))
+end
+function tensorproduct(A, IA, B, IB, IC=vcat(IA, IB))
+    return tensorproduct(A, tuple(IA...), B, tuple(IB...), tuple(IC...))
+end
 
 """
     tensorcopy(A, IA, IC = IA)
@@ -106,7 +109,7 @@ this is a special case of `tensorcontract` with no indices being contracted over
 This method checks whether the indices indeed specify a tensor product instead of
 a genuine contraction.
 """
-function tensorproduct(A, IA::Tuple, B, IB::Tuple, IC::Tuple = (IA..., IB...))
+function tensorproduct(A, IA::Tuple, B, IB::Tuple, IC::Tuple=(IA..., IB...))
     isempty(intersect(IA, IB)) || throw(IndexError("not a valid tensor product"))
-    tensorcontract(A, IA, B, IB, IC)
+    return tensorcontract(A, IA, B, IB, IC)
 end