Skip to content

Commit

Permalink
feat(KDP): adding options enums + comments
Browse files Browse the repository at this point in the history
  • Loading branch information
piotrlaczkowski committed Mar 12, 2024
1 parent 0ef86a7 commit 598e2e7
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion kdp/processor.py
Original file line number Diff line number Diff line change
Expand Up @@ -581,6 +581,7 @@ def build_preprocessor(self) -> tf.keras.Model:
)

# Preparing outputs
logger.info("Preparing outputs for the model")
self._prepare_outputs()

# building model
Expand Down Expand Up @@ -628,7 +629,8 @@ def plot_model(self) -> None:
"""Plotting model architecture.
Note:
This function requires graphviz to be installed on the system.
This function requires graphviz to be installed on the system
and pydot library (dependency in the dev group).
"""
logger.info("Plotting model")
return tf.keras.utils.plot_model(
Expand Down

0 comments on commit 598e2e7

Please sign in to comment.