fix notebook

slimgroup · Oct 6, 2022 · 1bb0e53 · 1bb0e53
1 parent c8e2d13
commit 1bb0e53
Show file tree

Hide file tree

Showing 3 changed files with 58 additions and 121 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "PhotoAcoustic"
 uuid = "86b14aa7-fcb7-4836-b4c7-056f45a9c77b"
 authors = ["rafaelorozco <[email protected]>", "Mathias Louboutin <[email protected]"]
-version = "0.2.0"
+version = "0.3.0"
 
 [deps]
 JUDI = "f3b833dc-6b2e-5b9c-b940-873ed6319979"

diff --git a/examples/notebooks/Flux_AD_Integration_Deep_Prior.ipynb b/examples/notebooks/Flux_AD_Integration_Deep_Prior.ipynb
diff --git a/src/judiPhoto.jl b/src/judiPhoto.jl
@@ -66,7 +66,7 @@ function propagate(J::judiPhoto{T, :forward}, q::AbstractArray{T}) where {T}
     # Set up Python model structure
     modelPy = devito_model(J.F.model, J.F.options)
     dtComp  = convert(Float32, modelPy."critical_dt")
-    nt = floor(Int, recGeometry.t[1] / dtComp) + 1
+    nt = length(0:dtComp:((recGeometry.nt[1]-1)*recGeometry.dt[1]))
 
     # Set up coordinates
     rec_coords = setup_grid(recGeometry, J.F.model.n)    # shifts rec coordinates by origin