Reintroduce minimal AbstractNetworkIterator interface code to express BP

.github/workflows/FormatCheck.yml

@@ -1,11 +1,14 @@
 name: "Format Check"
 
 on:
-  push:
-    branches:
-      - 'main'
-    tags: '*'
-  pull_request:
+  pull_request_target:
+    paths: ['**/*.jl']
+    types: [opened, synchronize, reopened, ready_for_review]
+
+permissions:
+  contents: read
+  actions: write
+  pull-requests: write
 
 jobs:
   format-check:

.pre-commit-config.yaml

@@ -1,5 +1,5 @@
 ci:
-  skip: [julia-formatter]
+  skip: [runic]
 
 repos:
 - repo: https://github.com/pre-commit/pre-commit-hooks
@@ -11,7 +11,7 @@ repos:
   - id: end-of-file-fixer
     exclude_types: [markdown]  # incompatible with Literate.jl
 
-- repo: "https://github.com/domluna/JuliaFormatter.jl"
-  rev: v2.1.6
+- repo: https://github.com/fredrikekre/runic-pre-commit
+  rev: v2.0.1
   hooks:
-  - id: "julia-formatter"
+    - id: runic

Project.toml

@@ -1,31 +1,48 @@
 name = "ITensorNetworksNext"
 uuid = "302f2e75-49f0-4526-aef7-d8ba550cb06c"
 authors = ["ITensor developers <[email protected]> and contributors"]
-version = "0.1.2"
+version = "0.1.11"
 
 [deps]
+AbstractTrees = "1520ce14-60c1-5f80-bbc7-55ef81b5835c"
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
 BackendSelection = "680c2d7c-f67a-4cc9-ae9c-da132b1447a5"
 DataGraphs = "b5a273c3-7e6c-41f6-98bd-8d7f1525a36a"
 Dictionaries = "85a47980-9c8c-11e8-2b9f-f7ca1fa99fb4"
 Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6"
+ITensorBase = "4795dd04-0d67-49bb-8f44-b89c448a1dc7"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09"
 NamedDimsArrays = "60cbd0c0-df58-4cb7-918c-6f5607b73fde"
 NamedGraphs = "678767b0-92e7-4007-89e4-4527a8725b19"
 SimpleTraits = "699a6c99-e7fa-54fc-8d76-47d257e15c1d"
 SplitApplyCombine = "03a91e81-4c3e-53e1-a0a4-9c0c8f19dd66"
+TermInterface = "8ea1fca8-c5ef-4a55-8b96-4e9afe9c9a3c"
+TypeParameterAccessors = "7e5a90cf-f82e-492e-a09b-e3e26432c138"
+WrappedUnions = "325db55a-9c6c-5b90-b1a2-ec87e7a38c44"
+
+[weakdeps]
+TensorOperations = "6aa20fa7-93e2-5fca-9bc0-fbd0db3c71a2"
+
+[extensions]
+ITensorNetworksNextTensorOperationsExt = "TensorOperations"
 
 [compat]
-Adapt = "4.3.0"
+AbstractTrees = "0.4.5"
+Adapt = "4.3"
 BackendSelection = "0.1.6"
 DataGraphs = "0.2.7"
 Dictionaries = "0.4.5"
 Graphs = "1.13.1"
+ITensorBase = "0.2.14"
 LinearAlgebra = "1.10"
 MacroTools = "0.5.16"
-NamedDimsArrays = "0.7.13"
-NamedGraphs = "0.6.9"
+NamedDimsArrays = "0.8"
+NamedGraphs = "0.6.9, 0.7"
 SimpleTraits = "0.9.5"
 SplitApplyCombine = "1.2.3"
+TensorOperations = "5.3.1"
+TermInterface = "2"
+TypeParameterAccessors = "0.4.4"
+WrappedUnions = "0.3"
 julia = "1.10"

docs/make.jl

@@ -2,23 +2,23 @@ using ITensorNetworksNext: ITensorNetworksNext
 using Documenter: Documenter, DocMeta, deploydocs, makedocs
 
 DocMeta.setdocmeta!(
-  ITensorNetworksNext, :DocTestSetup, :(using ITensorNetworksNext); recursive=true
+    ITensorNetworksNext, :DocTestSetup, :(using ITensorNetworksNext); recursive = true
 )
 
 include("make_index.jl")
 
 makedocs(;
-  modules=[ITensorNetworksNext],
-  authors="ITensor developers <[email protected]> and contributors",
-  sitename="ITensorNetworksNext.jl",
-  format=Documenter.HTML(;
-    canonical="https://itensor.github.io/ITensorNetworksNext.jl",
-    edit_link="main",
-    assets=["assets/favicon.ico", "assets/extras.css"],
-  ),
-  pages=["Home" => "index.md", "Reference" => "reference.md"],
+    modules = [ITensorNetworksNext],
+    authors = "ITensor developers <[email protected]> and contributors",
+    sitename = "ITensorNetworksNext.jl",
+    format = Documenter.HTML(;
+        canonical = "https://itensor.github.io/ITensorNetworksNext.jl",
+        edit_link = "main",
+        assets = ["assets/favicon.ico", "assets/extras.css"],
+    ),
+    pages = ["Home" => "index.md", "Reference" => "reference.md"],
 )
 
 deploydocs(;
-  repo="github.com/ITensor/ITensorNetworksNext.jl", devbranch="main", push_preview=true
+    repo = "github.com/ITensor/ITensorNetworksNext.jl", devbranch = "main", push_preview = true
 )

docs/make_index.jl

@@ -2,20 +2,20 @@ using Literate: Literate
 using ITensorNetworksNext: ITensorNetworksNext
 
 function ccq_logo(content)
-  include_ccq_logo = """
+    include_ccq_logo = """
     ```@raw html
     <img class="display-light-only" src="assets/CCQ.png" width="20%" alt="Flatiron Center for Computational Quantum Physics logo."/>
     <img class="display-dark-only" src="assets/CCQ-dark.png" width="20%" alt="Flatiron Center for Computational Quantum Physics logo."/>
     ```
     """
-  content = replace(content, "{CCQ_LOGO}" => include_ccq_logo)
-  return content
+    content = replace(content, "{CCQ_LOGO}" => include_ccq_logo)
+    return content
 end
 
 Literate.markdown(
-  joinpath(pkgdir(ITensorNetworksNext), "examples", "README.jl"),
-  joinpath(pkgdir(ITensorNetworksNext), "docs", "src");
-  flavor=Literate.DocumenterFlavor(),
-  name="index",
-  postprocess=ccq_logo,
+    joinpath(pkgdir(ITensorNetworksNext), "examples", "README.jl"),
+    joinpath(pkgdir(ITensorNetworksNext), "docs", "src");
+    flavor = Literate.DocumenterFlavor(),
+    name = "index",
+    postprocess = ccq_logo,
 )

docs/make_readme.jl

@@ -2,20 +2,20 @@ using Literate: Literate
 using ITensorNetworksNext: ITensorNetworksNext
 
 function ccq_logo(content)
-  include_ccq_logo = """
+    include_ccq_logo = """
     <picture>
       <source media="(prefers-color-scheme: dark)" width="20%" srcset="docs/src/assets/CCQ-dark.png">
       <img alt="Flatiron Center for Computational Quantum Physics logo." width="20%" src="docs/src/assets/CCQ.png">
     </picture>
     """
-  content = replace(content, "{CCQ_LOGO}" => include_ccq_logo)
-  return content
+    content = replace(content, "{CCQ_LOGO}" => include_ccq_logo)
+    return content
 end
 
 Literate.markdown(
-  joinpath(pkgdir(ITensorNetworksNext), "examples", "README.jl"),
-  joinpath(pkgdir(ITensorNetworksNext));
-  flavor=Literate.CommonMarkFlavor(),
-  name="README",
-  postprocess=ccq_logo,
+    joinpath(pkgdir(ITensorNetworksNext), "examples", "README.jl"),
+    joinpath(pkgdir(ITensorNetworksNext));
+    flavor = Literate.CommonMarkFlavor(),
+    name = "README",
+    postprocess = ccq_logo,
 )

examples/README.jl

@@ -1,5 +1,5 @@
 # # ITensorNetworksNext.jl
-# 
+#
 # [![Stable](https://img.shields.io/badge/docs-stable-blue.svg)](https://itensor.github.io/ITensorNetworksNext.jl/stable/)
 # [![Dev](https://img.shields.io/badge/docs-dev-blue.svg)](https://itensor.github.io/ITensorNetworksNext.jl/dev/)
 # [![Build Status](https://github.com/ITensor/ITensorNetworksNext.jl/actions/workflows/Tests.yml/badge.svg?branch=main)](https://github.com/ITensor/ITensorNetworksNext.jl/actions/workflows/Tests.yml?query=branch%3Amain)

ext/ITensorNetworksNextTensorOperationsExt/ITensorNetworksNextTensorOperationsExt.jl

@@ -0,0 +1,16 @@
+module ITensorNetworksNextTensorOperationsExt
+
+using BackendSelection: @Algorithm_str, Algorithm
+using NamedDimsArrays: inds
+using ITensorNetworksNext: ITensorNetworksNext, contraction_sequence_to_expr
+using TensorOperations: TensorOperations, optimaltree
+
+function ITensorNetworksNext.contraction_sequence(::Algorithm"optimal", tn::Vector{<:AbstractArray})
+    network = collect.(inds.(tn))
+    #Converting dims to Float64 to minimize overflow issues
+    inds_to_dims = Dict(i => Float64(length(i)) for i in unique(reduce(vcat, network)))
+    seq, _ = optimaltree(network, inds_to_dims)
+    return contraction_sequence_to_expr(seq)
+end
+
+end

src/ITensorNetworksNext.jl

@@ -1,6 +1,14 @@
 module ITensorNetworksNext
 
+include("lazynameddimsarrays.jl")
 include("abstracttensornetwork.jl")
 include("tensornetwork.jl")
+include("contract_network.jl")
+include("abstract_problem.jl")
+include("iterators.jl")
+
+include("beliefpropagation/abstractbeliefpropagationcache.jl")
+include("beliefpropagation/beliefpropagationcache.jl")
+include("beliefpropagation/beliefpropagationproblem.jl")
 
 end

src/abstract_problem.jl

@@ -0,0 +1 @@
+abstract type AbstractProblem end