From 3c45ae1d7b67e0cac912608811d295b652bef81c Mon Sep 17 00:00:00 2001 From: LordMathis Date: Wed, 13 Dec 2023 23:34:58 +0100 Subject: [PATCH] Change auth env vars --- mlflow_plugin_proxy_auth/proxy_auth_header_provider.py | 4 ++-- setup.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) 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={