Convert n_steps from DeviceArray to int

Author: rlouf

aesara/link/jax/dispatch/scan.py

@@ -33,9 +33,8 @@ def scan(*outer_inputs):
         # + outer_in_shared
         # + outer_in_nit_sot
         # + outer_in_non_seqs
-        n_steps = scan_args.n_steps
+        n_steps = scan_args.n_steps.item()  # JAX needs an int
         seqs = scan_args.outer_in_seqs
-
         # TODO: mit_mots
         mit_mot_in_slices = []