Mathis
/
mlflow-plugin-proxy-auth
mirror of https://github.com/LordMathis/mlflow-plugin-proxy-auth.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlflow_plugin_proxy_auth/proxy_auth_header_provider.py
View File

@ -9,8 +9,8 @@ from requests.auth import AuthBase
class ProxyAuthProvider(RequestAuthProvider): class ProxyAuthProvider(RequestAuthProvider):
def __init__(self): def __init__(self):
self.username = os.getenv("MLFLOW_TRACKING_USERNAME") self.username = os.getenv("MLFLOW_PROXY_USERNAME")
self.password = os.getenv("MLFLOW_TRACKING_PASSWORD") self.password = os.getenv("MLFLOW_PROXY_PASSWORD")
def get_name(self): def get_name(self):
return "proxy_auth_provider" return "proxy_auth_provider"

2
setup.py
View File

@ -2,7 +2,7 @@ from setuptools import find_packages, setup
setup( setup(
name="mlflow-plugin-proxy-auth", name="mlflow-plugin-proxy-auth",
version="0.0.3", version="0.0.4",
packages=find_packages(), packages=find_packages(),
install_requires=["mlflow"], install_requires=["mlflow"],
entry_points={ entry_points={