Update DeterministicParticleFlowControl.py

Author: dimitra-maoutsa

DeterministicParticleFlowControl/DeterministicParticleFlowControl.py

@@ -20,7 +20,7 @@
 
 
 class DPFC:
-    def __init__(self,t1,t2,y1,y2,f,g,N,M,reweight=False, U=None,dens_est='nonparametric',reject=True,plotting=True,kern='RBF',f_true=None,brown_bridge=False):
+    def __init__(self,t1,t2,y1,y2,f,g,N,M,reweight=False, U=None,dens_est='nonparametric',reject=True,kern='RBF',f_true=None,brown_bridge=False):
         """
         Deterministic particle flow control - class initialising function
         t1: starting time point
@@ -245,7 +245,7 @@ def forward_sampling_Otto(self):
             if self.reweight == True:
               if ti>0:
 
-                  W[:,0] = np.exp(self.U(self.Z[:,:,ti])*self.dt ) #-1                   
+                  W[:,0] = np.exp(self.U(self.Z[:,:,ti],tt)*self.dt ) #-1                   
                   W = W/np.sum(W)       
 
                   ###REWEIGHT