From ffc6da1e96c9d6453d4adb66be6ec25f9f1ca346 Mon Sep 17 00:00:00 2001
From: HuaiMing <huaimingyuuu@gmail.com>
Date: Fri, 12 Aug 2022 11:13:34 +0800
Subject: [PATCH] revise NM struct

---
 src/Algorithm/Algorithm.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/Algorithm/Algorithm.jl b/src/Algorithm/Algorithm.jl
index 5c9c201..70f8578 100644
--- a/src/Algorithm/Algorithm.jl
+++ b/src/Algorithm/Algorithm.jl
@@ -156,10 +156,10 @@ Optimization algorithm: DE.
 DDPG(;max_episode::Int=500, layer_num::Int=3, layer_dim::Int=200, seed::Number=1234) =
     DDPG(max_episode, layer_num, layer_dim, StableRNG(seed))
 
-struct NM{T<:AbstractVector,N<:Number} <: AbstractAlgorithm
+struct NM{N<:Number} <: AbstractAlgorithm
     max_episode::Int
     p_num::Int
-    ini_state::Union{T, Missing}
+    ini_state::Union{Vector{ComplexF64}, Missing} 
     ar::N
     ae::N
     ac::N