added arguments to azuremldataasset in order to be able to specify th…

…e path within azureml where the dataset is saved

kedro_azureml/datasets/asset_dataset.py

@@ -77,6 +77,8 @@ def __init__(
         self,
         azureml_dataset: str,
         dataset: Union[str, Type[AbstractDataSet], Dict[str, Any]],
+        datastore: str = "workspaceblobstore",
+        azureml_root_dir: str = "kedro_azureml",  # maybe combine with root_dir?
         root_dir: str = "data",
         filepath_arg: str = "filepath",
         azureml_type: AzureMLDataAssetType = "uri_folder",
@@ -93,6 +95,8 @@ def __init__(
         """
         super().__init__(dataset=dataset, root_dir=root_dir, filepath_arg=filepath_arg)
 
+        self._azureml_root_dir = azureml_root_dir
+        self._datastore = datastore
         self._azureml_dataset = azureml_dataset
         self._version = version
         # 1 entry for load version, 1 for save version

kedro_azureml/generator.py

@@ -177,7 +177,11 @@ def _get_output(self, name):
                     "AzureMLAssetDataSets with azureml_type 'uri_file' cannot be used as outputs"
                 )
             # TODO: add versioning
-            return Output(type=ds._azureml_type, name=ds._azureml_dataset)
+            return Output(
+                type=ds._azureml_type,
+                name=ds._azureml_dataset,
+                path=f"azureml://datastores/{ds._datastore}/paths/{ds._azureml_root_dir}/{ds.resolve_save_version()}",
+            )
         else:
             return Output(type="uri_folder")