Skip to content
Merged
Changes from 1 commit
Commits
Show all changes
17 commits
Select commit Hold shift + click to select a range
648d0a9
Bug 3323988: Regex fix and indices correction for model download
kshitij-microsoft Jul 2, 2024
c21f605
Merge branch 'main' of github.com:Azure/azure-sdk-for-python
kshitij-microsoft Jul 4, 2024
331d7e3
Merge branch 'main' of github.com:Azure/azure-sdk-for-python
kshitij-microsoft Jul 5, 2024
f125053
fixing test case
kshitij-microsoft Jul 17, 2024
a5aa64a
Merge branch 'main' of github.com:Azure/azure-sdk-for-python
kshitij-microsoft Jul 23, 2024
921e805
adding mlflow tracking uri func
kshitij-microsoft Jul 23, 2024
d4785cd
Merge branch 'main' of github.com:Azure/azure-sdk-for-python
kshitij-microsoft Jul 31, 2024
e647204
passing service context to azureml mlflow
kshitij-microsoft Aug 2, 2024
2ed6544
Merge branch 'main' of github.com:Azure/azure-sdk-for-python
kshitij-microsoft Aug 28, 2024
a03c7a3
final flow APC complete
kshitij-microsoft Sep 5, 2024
357698f
Merge branch 'main' of github.com:Azure/azure-sdk-for-python
kshitij-microsoft Sep 6, 2024
d688c3c
Merge branch 'main' of https://github.com/Azure/azure-sdk-for-python
kshitij-microsoft Sep 24, 2024
3dbb2d2
modify host_url
kshitij-microsoft Sep 25, 2024
58a96dc
fixing unit test cases
kshitij-microsoft Sep 27, 2024
ae32d9d
Merge branch 'main' of https://github.com/Azure/azure-sdk-for-python
kshitij-microsoft Sep 30, 2024
0ab7bc4
changing mock for urlparse
kshitij-microsoft Oct 1, 2024
5a90da5
fixing the log msg
kshitij-microsoft Oct 24, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
adding mlflow tracking uri func
  • Loading branch information
kshitij-microsoft committed Jul 23, 2024
commit 921e80593682779185f8d0fd17bb849fcc920c2e
25 changes: 23 additions & 2 deletions sdk/ml/azure-ai-ml/azure/ai/ml/entities/_workspace/workspace.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,13 @@
from azure.ai.ml.entities._resource import Resource
from azure.ai.ml.entities._util import find_field_in_override, load_from_dict
from azure.ai.ml.entities._workspace.serverless_compute import ServerlessComputeSettings
from azure.ai.ml.exceptions import ErrorCategory, ErrorTarget, ValidationErrorType, ValidationException
from azure.ai.ml.exceptions import (
ErrorCategory,
ErrorTarget,
UserErrorException,
ValidationErrorType,
ValidationException,
)

from .customer_managed_key import CustomerManagedKey
from .feature_store_settings import FeatureStoreSettings
Expand Down Expand Up @@ -222,7 +228,22 @@ def mlflow_tracking_uri(self) -> Optional[str]:
:return: Returns mlflow tracking uri of the workspace.
:rtype: str
"""
return self._mlflow_tracking_uri
# if _with_auth:
# module_logger.warning(
# "'_with_auth' is deprecated and will be removed in a future release. ")

try:
from azureml.mlflow import get_mlflow_tracking_uri_v2

return get_mlflow_tracking_uri_v2(self)
except ImportError as e:
error_msg = (
"azureml.mlflow could not be imported. "
"Please ensure that 'azureml-mlflow' has been installed in the current python environment."
)
raise UserErrorException(error_msg) from e

# return self._mlflow_tracking_uri

def dump(self, dest: Union[str, PathLike, IO[AnyStr]], **kwargs: Any) -> None:
"""Dump the workspace spec into a file in yaml format.
Expand Down