Skip to content

Commit

Permalink
Fix docstr.
Browse files Browse the repository at this point in the history
  • Loading branch information
rousseab committed Apr 5, 2024
1 parent bc84515 commit 2dfcd15
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions crystal_diffusion/callbacks/callback_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,8 @@
def create_all_callbacks(hyper_params: Dict[AnyStr, Any], output_directory: str, verbose: bool) -> Dict[str, Callback]:
"""Create all callbacks.
This method leverages the global dictionary CALLBACK_DICTIONARY which should be used to
register all available callbacks and provide a standardized interface to intialize the callbacks.
This method leverages the global dictionary OPTIONAL_CALLBACK_DICTIONARY which should be used to
register all available optional callbacks and provide a standardized interface to intialize these callbacks.
The instantiation methods can define sane defaults or hardcode constant choices.
Expand All @@ -26,6 +26,7 @@ def create_all_callbacks(hyper_params: Dict[AnyStr, Any], output_directory: str,
Returns:
all_callbacks_dict: a dictionary of instantiated callbacks with relevant names as keys.
"""
# We always need a progress bar.
all_callbacks_dict = dict(progress_bar=CustomProgressBar())

for callback_name, instantiate_callback in OPTIONAL_CALLBACK_DICTIONARY.items():
Expand Down

0 comments on commit 2dfcd15

Please sign in to comment.