Merge branch 'ml' of https://github.com/jeertmans/DiffeRT2d into ml

jeertmans · May 21, 2024 · 8a34ebf · 8a34ebf
2 parents bb86d96 + ca8e1e9
commit 8a34ebf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/differt2d/models/cost20120.py b/differt2d/models/cost20120.py
@@ -138,10 +138,10 @@ def __call__(
     ) -> tuple[Float[Array, " "], Float[Array, "{self.order}+2 2"]]:
         # Generate new state with path starting at TX
 
-        #jax.debug.print("1. Initial state of path gen = {state}", state=init_state)
+        # jax.debug.print("1. Initial state of path gen = {state}", state=init_state)
         init_state = self.cell(tx, init_state)
 
-        #jax.debug.print("2. Initial state of path gen = {state}", state=init_state)
+        # jax.debug.print("2. Initial state of path gen = {state}", state=init_state)
         @jax.jit
         @jaxtyped(typechecker=typechecker)
         def scan_fn(