fix: prints

based-robotics · Sep 17, 2024 · 75bbc9e · 75bbc9e
1 parent a24ad63
commit 75bbc9e
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/jaxadi/_stages.py b/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
 
 

diff --git a/jaxadi/_translate.py b/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