Skip to content

Commit 6d68f28

Browse files
Clarify MLflow catalog.yml file context in docs (#5389)
* Clarify MLflow catalog.yml file context in docs Add `# conf/base/catalog.yml` comments to YAML code snippets in the MLflow integration docs to clarify where readers should place the configuration. Closes #5384 Signed-off-by: Mridul Tailor <71081929+MridulTailor@users.noreply.github.com> * Apply suggestion to delete intermediate blank line Signed-off-by: Deepyaman Datta <deepyaman.datta@utexas.edu> Signed-off-by: Deepyaman Datta <deepyaman.datta@utexas.edu> * Add catalog.yml comment to model registry diff snippet Signed-off-by: Mridul Tailor <71081929+MridulTailor@users.noreply.github.com> --------- Signed-off-by: Mridul Tailor <71081929+MridulTailor@users.noreply.github.com> Signed-off-by: Deepyaman Datta <deepyaman.datta@utexas.edu> Co-authored-by: Deepyaman Datta <deepyaman.datta@utexas.edu>
1 parent eb6be6b commit 6d68f28

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

docs/integrations-and-plugins/mlflow.md

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,8 @@ For example, if you have a dataset corresponding to a scikit-learn model,
142142
you can change it as follows:
143143

144144
```diff
145-
regressor:
145+
# conf/base/catalog.yml
146+
regressor:
146147
- type: pickle.PickleDataset
147148
- filepath: data/06_models/regressor.pickle
148149
- versioned: true
@@ -158,6 +159,7 @@ If you also want to _register_ it
158159
you can add a `registered_model_name` parameter:
159160

160161
```yaml
162+
# conf/base/catalog.yml
161163
regressor:
162164
type: kedro_mlflow.io.models.MlflowModelTrackingDataset
163165
flavor: mlflow.sklearn
@@ -173,6 +175,7 @@ To load a model from a specific run, you can specify the `run_id`.
173175
For that, you can make use of {ref}`runtime parameters <runtime-params>`:
174176

175177
```yaml
178+
# conf/base/catalog.yml
176179
# Add the intermediate datasets to run the inference pipeline
177180
X_test:
178181
type: pandas.ParquetDataset

0 commit comments

Comments
 (0)