fix: prints

Author: mattephi

jaxadi/_stages.py

@@ -115,9 +115,7 @@ def stage_generator(func: Function) -> str:
             workers[o_idx[0]] = OP_JAX_VALUE_DICT[op].format(workers[i_idx[0]])
         else:
             raise Exception("Unknown CasADi operation: " + str(op))
-        print(sum(len(s) for s in workers))
 
-    print("finished stages")
     return stages
 
 

jaxadi/_translate.py

@@ -7,7 +7,6 @@ def translate(func: Function, add_jit=False, add_import=False, num_threads=1) ->
 
     # get the shapes of input and output
     out_shapes = [func.size_out(i) for i in range(n_out)]
-    print(out_shapes)
     stages = stage_generator(func)
     stages = squeeze(stages, num_threads=num_threads)
     # get information about casadi function