modalities.logging_broker.subscriber_impl package

Submodules

modalities.logging_broker.subscriber_impl.progress_subscriber module

class modalities.logging_broker.subscriber_impl.progress_subscriber.DummyProgressSubscriber

Bases: MessageSubscriberIF[ProgressUpdate]

consume_dict(message_dict)

Parameters:

message_dict (dict[str, Any])

consume_message(message)

Parameters:

message (Message[ProgressUpdate])

class modalities.logging_broker.subscriber_impl.progress_subscriber.RichProgressSubscriber(train_split_num_steps, eval_splits_num_steps)

Bases: MessageSubscriberIF[ProgressUpdate]

A subscriber object for the RichProgress observable.

Parameters:

• train_split_num_steps (dict[str, tuple[int, int]])

• eval_splits_num_steps (dict[str, int])

consume_dict(message_dict)

Parameters:

message_dict (dict[str, Any])

consume_message(message)

Consumes a message from a message broker.

Parameters:

message (Message[ProgressUpdate])

classmethod register_live_display(live_display)

Only one instance of rich.live.Live can run at the same time. Therefore we use a singleton approach to have only one active,

by storing the active reference as class-field here.

Parameters:

live_display (Live)

modalities.logging_broker.subscriber_impl.results_subscriber module

modalities.logging_broker.subscriber_impl.subscriber_factory module

Module contents