diff --git a/mlflow_plugin_proxy_auth/proxy_auth_header_provider.py b/mlflow_plugin_proxy_auth/proxy_auth_header_provider.py index 4f983c4..dd8e453 100644 --- a/mlflow_plugin_proxy_auth/proxy_auth_header_provider.py +++ b/mlflow_plugin_proxy_auth/proxy_auth_header_provider.py @@ -9,8 +9,8 @@ from requests.auth import AuthBase class ProxyAuthProvider(RequestAuthProvider): def __init__(self): - self.username = os.getenv("MLFLOW_TRACKING_USERNAME") - self.password = os.getenv("MLFLOW_TRACKING_PASSWORD") + self.username = os.getenv("MLFLOW_PROXY_USERNAME") + self.password = os.getenv("MLFLOW_PROXY_PASSWORD") def get_name(self): return "proxy_auth_provider" diff --git a/setup.py b/setup.py index 9334080..56122bf 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ from setuptools import find_packages, setup setup( name="mlflow-plugin-proxy-auth", - version="0.0.3", + version="0.0.4", packages=find_packages(), install_requires=["mlflow"], entry_points={