Fix rand on TensorNetwork

bsc-quantic · Sep 13, 2024 · 58e44f1 · 58e44f1
1 parent 72ea6d6
commit 58e44f1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/TensorNetwork.jl b/src/TensorNetwork.jl
@@ -721,6 +721,10 @@ function Base.rand(
     return TensorNetwork(tensors)
 end
 
+function Base.rand(::Type{TensorNetwork}, n::Integer, regularity::Integer; kwargs...)
+    return rand(Random.default_rng(), TensorNetwork, n, regularity; kwargs...)
+end
+
 function Serialization.serialize(s::AbstractSerializer, obj::TensorNetwork)
     Serialization.writetag(s.io, Serialization.OBJECT_TAG)
     return serialize(s, tensors(obj))