-
Notifications
You must be signed in to change notification settings - Fork 34
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
FIX #58 - Move MlflowPipelineHook args to PipelineML class #60
Conversation
This will close #53 btw. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
As pipeline_ml
is user-facing I think it should have its own docstring. Apart from documentation comments, it looks good 👌
Just one more question: I didn't check but does this changeset require any updates to the documentation?
kedro_mlflow/pipeline/pipeline_ml.py
Outdated
"""Store all necessary information for calling mlflow.log_model in the pipeline. | ||
|
||
Args: | ||
nodes (Iterable[Union[Node,): The `node`s of the training pipeline. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
- The type of nodes needs updating.
- The order of arguments in docstring is wrong.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I did not notice that the autodocstring extension changed the order of the argument in the docstring. Idon't know why, maybe because of *args not being the last argument.
kedro_mlflow/pipeline/pipeline_ml.py
Outdated
- a path to an "environment.yml" : data is loaded and used as they are | ||
- a Dict : used as the environment | ||
- None: a base conda environment with your current python version and your project version at training time. | ||
Defaults to None. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
- This should be indented with one more tab
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Well spotted. I have also refactored the docstring to make shorter lines length for readibility.
If you put |
45e0750
to
99e4ad6
Compare
…lineHook` to `PipelineML` class
99e4ad6
to
7b08011
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
All good. Merging.
conda_env
andmodel_name
arguments fromMlflowPipelineHook
toPipelineML
class #58: moveMlflowPipelineHook
args toPipelineML
classRemoveflatten_dict_param
fromMlflowNodeHook
#59: moveMlflowNodeHook
args toKedroMlflowConfig
classI have changed my mind: #59 is much trickier than it seems because of the need to import
get_mlflow_config
inside the hook which has a lot of side effects. For the sake of clarity, let's merge only the fix for #58 first. I'll open a separated PR for #59.