Change auth env vars

This commit is contained in:
LordMathis 2023-12-13 23:34:58 +01:00
parent 77c318b32a
commit 3c45ae1d7b
2 changed files with 3 additions and 3 deletions

View File

@ -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"

View File

@ -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={