Skip to content

Commit

Permalink
feat(workshop): clean up for workshop
Browse files Browse the repository at this point in the history
  • Loading branch information
guillaume-chervet committed Dec 17, 2023
1 parent 92f46f6 commit bc0b6be
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 3 deletions.
2 changes: 1 addition & 1 deletion train/azureml_run_pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,7 @@ def azureml_pipeline(

pipeline_job = azureml_pipeline(
pdfs_input_data=Input(
path="azureml:cats_dogs_others:1", type=AssetTypes.URI_FOLDER
path="azureml:cats_dogs_others:1", type=AssetTypes.URI_FOLDER,
),
labels_input_data=Input(
path="azureml:cats_dogs_others_labels:1", type=AssetTypes.URI_FOLDER
Expand Down
3 changes: 1 addition & 2 deletions train/extraction/command.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,8 +24,7 @@
console_output = f"""
number_files_input: {result.number_files_input}
number_images_output: {result.number_images_output}
computed_hash: {computed_hash}
"""
computed_hash: {computed_hash}"""

mlflow.log_metric("number_files_input", result.number_files_input)
mlflow.log_metric("number_images_output", result.number_images_output)

0 comments on commit bc0b6be

Please sign in to comment.