Skip to content

Commit

Permalink
Log nit
Browse files Browse the repository at this point in the history
  • Loading branch information
brentyi committed Sep 29, 2024
1 parent 2f5d916 commit a2d60e6
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions src/jaxls/_factor_graph.py
Original file line number Diff line number Diff line change
Expand Up @@ -222,7 +222,7 @@ def _sort_key(x: Any) -> str:
for group_key in sorted(factors_from_group.keys(), key=_sort_key):
group = factors_from_group[group_key]
logger.info(
"Group with factors={}, variables={}: {}",
"Vectorizing group with {} factors, {} variables each: {}",
count_from_group[group_key],
group[0].num_variables,
group[0].compute_residual.__name__,
Expand Down Expand Up @@ -278,7 +278,6 @@ def _sort_key(x: Any) -> str:
indptr=cast(jax.Array, csr_indptr),
shape=(residual_dim_sum, tangent_dim_sum),
)
logger.info("Done!")
return FactorGraph(
stacked_factors=tuple(stacked_factors),
factor_counts=tuple(factor_counts),
Expand Down

0 comments on commit a2d60e6

Please sign in to comment.