Merge pull request #35 from JoeyT1994/random

Add RNG and eltype suppport for random_itensornetworks
JoeyT1994 · May 29, 2024 · 5d728f6 · 5d728f6
2 parents d9aadc5 + 856d12e
commit 5d728f6
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/Project.toml b/Project.toml
@@ -17,7 +17,7 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 Dictionaries = "0.4.2"
 Glob = "1.3.1"
 Graphs = "1.8"
-ITensorNetworks = "0.11.8"
+ITensorNetworks = "0.11.10"
 ITensors = "0.5, 0.6"
 NamedGraphs = "0.6"
 Random = "1.8"

diff --git a/src/elementary_functions.jl b/src/elementary_functions.jl
@@ -4,6 +4,7 @@ using NamedGraphs.GraphsExtensions:
   random_bfs_tree, rem_edges, add_edges, leaf_vertices, undirected_graph
 using ITensors: dim, commoninds
 using ITensorNetworks: IndsNetwork, underlying_graph
+using Random: AbstractRNG
 
 default_c_value() = 1.0
 default_a_value() = 0.0
@@ -191,6 +192,16 @@ function polynomial_itensornetwork(
   return ψ
 end
 
+function random_itensornetwork(rng::AbstractRNG, eltype::Type, s::IndsNetworkMap; kwargs...)
+  return ITensorNetworkFunction(
+    random_tensornetwork(rng, eltype, indsnetwork(s); kwargs...), s
+  )
+end
+
+function random_itensornetwork(rng::AbstractRNG, s::IndsNetworkMap; kwargs...)
+  return ITensorNetworkFunction(random_tensornetwork(rng, indsnetwork(s); kwargs...), s)
+end
+
 function random_itensornetwork(s::IndsNetworkMap; kwargs...)
   return ITensorNetworkFunction(random_tensornetwork(indsnetwork(s); kwargs...), s)
 end