making the format consistant for the gt4py kernel

Author: supreethms1809

swm_python/swm_cartesian.py

@@ -271,7 +271,7 @@ def calc_uold(
         uold: gtscript.Field[dtype]
     ):
         with computation(PARALLEL), interval(...):
-            uold[...] = u + alpha * (unew - 2 * u + uold)
+            uold = u + alpha * (unew - 2 * u + uold)
 
     @gtscript.stencil(backend=cartesian_backend)
     def calc_vold(
@@ -281,13 +281,13 @@ def calc_vold(
         vold: gtscript.Field[dtype]
     ):
         with computation(PARALLEL), interval(...):
-            vold[...] = v + alpha * (vnew - 2 * v + vold)
+            vold = v + alpha * (vnew - 2 * v + vold)
 
     time = 0.0
     # Main time loop
     for ncycle in range(ITMAX):
         if((ncycle%100==0) & (VIS==False)):
-            print("cycle number ", ncycle)
+            print(f"cycle number{ncycle} and gt4py type {gt4py_type}")
         # Calculate cu, cv, z, and h
         calc_h(p=p_gt, u=u_gt, v=v_gt, h=h_gt, origin=(0,0,0), domain=(nx,ny,nz)) 
         h = h_gt.asnumpy()