Added steps to register dataset and the environment

cv/aml-cli-v2/mlops/azureml/train/train-env.yaml

@@ -1,6 +1,5 @@
 $schema: https://azuremlschemas.azureedge.net/latest/environment.schema.json
 name: nvidia_pytorch
-version: 22.04-py3
 build:
   path: ../../../data-science/src/environment/
 tags:

cv/aml-cli-v2/mlops/azureml/train/train.yaml

@@ -95,7 +95,7 @@ outputs:
 
 code: ../../../data-science/src
 
-environment: azureml:azureml:nvidia_pytorch:22.04-py3
+environment: azureml:azureml:nvidia_pytorch@latest
 
 command: >-
   python train.py

cv/aml-cli-v2/mlops/devops-pipelines/deploy-model-training-pipeline.yml

@@ -44,6 +44,15 @@ stages:
           script: |
             echo  '$(System.DefaultWorkingDirectory)'
             tree $(System.DefaultWorkingDirectory) /f
+      - template: templates/${{ variables.version }}/register-environment.yml@mlops-templates
+        parameters:
+          build_type: docker
+          environment_name: nvidia_pytorch # Not used for docker builds
+          environment_file: cv/aml-cli-v2/mlops/azureml/train/train-env.yml       
+      - template: templates/${{ variables.version }}/register-dataset.yml@mlops-templates
+        parameters:
+          build_type: docker
+          environment_file: cv/aml-cli-v2/mlops/azureml/train/create_stanford_dogs_dataset.yaml     
       - template: templates/${{ variables.version }}/run-pipeline.yml@mlops-templates
         parameters: 
           pipeline_file: cv/aml-cli-v2/mlops/azureml/train/pipeline.yaml