fixes from review

Author: ayushinav

docs/src/tutorials/dgm.md

@@ -47,7 +47,7 @@ u(t, 1) & = 0
 ```
 
 ### Copy- Pasteable code
-```julia
+```@example
 using NeuralPDE
 using ModelingToolkit, Optimization, OptimizationOptimisers
 import Lux: tanh, identity
@@ -58,15 +58,15 @@ using MethodOfLines, OrdinaryDiffEq
 @parameters x t
 @variables u(..)
 
-Dt= Differential(t)
-Dx= Differential(x)
-Dxx= Dx^2
-α = 0.05;
+Dt = Differential(t)
+Dx = Differential(x)
+Dxx = Dx^2
+α = 0.05
 # Burger's equation
-eq= Dt(u(t,x)) + u(t,x) * Dx(u(t,x)) - α * Dxx(u(t,x)) ~ 0 
+eq = Dt(u(t,x)) + u(t,x) * Dx(u(t,x)) - α * Dxx(u(t,x)) ~ 0 
 
 # boundary conditions
-bcs= [
+bcs = [
     u(0.0, x) ~ - sin(π*x),
     u(t, -1.0) ~ 0.0,
     u(t, 1.0) ~ 0.0
@@ -75,7 +75,7 @@ bcs= [
 domains = [t ∈ Interval(0.0, 1.0), x ∈ Interval(-1.0, 1.0)]
 
 # MethodOfLines, for FD solution
-dx= 0.01
+dx = 0.01
 order = 2
 discretization = MOLFiniteDifference([x => dx], t, saveat = 0.01)
 @named pde_system = PDESystem(eq, bcs, domains, [t, x], [u(t,x)])
@@ -87,13 +87,13 @@ xs = sol[x]
 u_MOL = sol[u(t,x)]
 
 # NeuralPDE, using Deep Galerkin Method
-strategy = QuasiRandomTraining(256, minibatch= 32);
-    discretization= DeepGalerkin(2, 1, 50, 5, tanh, tanh, identity, strategy);
-@named pde_system = PDESystem(eq, bcs, domains, [t, x], [u(t,x)]);
-prob = discretize(pde_system, discretization);
-global iter = 0;
+strategy = QuasiRandomTraining(256, minibatch= 32)
+discretization = DeepGalerkin(2, 1, 50, 5, tanh, tanh, identity, strategy)
+@named pde_system = PDESystem(eq, bcs, domains, [t, x], [u(t,x)])
+prob = discretize(pde_system, discretization)
+global iter = 0
 callback = function (p, l)
-    global iter += 1;
+    global iter += 1
     if iter%20 == 0
         println("$iter => $l")
     end
@@ -103,11 +103,11 @@ end
 res = Optimization.solve(prob, Adam(0.1); callback = callback, maxiters = 100)
 prob = remake(prob, u0 = res.u)
 res = Optimization.solve(prob, Adam(0.01); callback = callback, maxiters = 500)
-phi = discretization.phi;
+phi = discretization.phi
 
 u_predict= [first(phi([t, x], res.minimizer)) for t in ts, x in xs]
 
-diff_u = abs.(u_predict .- u_MOL);
+diff_u = abs.(u_predict .- u_MOL)
 
 using Plots
 p1 = plot(tgrid, xgrid, u_MOL', linetype = :contourf, title = "FD");