treeflow.vi.progress_bar module

class treeflow.vi.progress_bar.ProgressBarFunc(*args, **kwargs)

Bases: Protocol

class treeflow.vi.progress_bar.ProgressBarTraceFunctionContextManager(tqdm: tqdm | None, trace_fn: Callable)

Bases: object

treeflow.vi.progress_bar.update_trace_fn(mtq: MinimizeTraceableQuantities, trace_fn: Callable, tqdm_instance: tqdm, update_step: int = 10)

treeflow.vi.progress_bar.make_progress_bar_trace_fn(trace_fn: Callable, num_steps: int, progress_bar: ProgressBarFunc | bool = True, update_step: int = 10)