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

Add docstrings

This commit is contained in:
LordMathis 2023-12-17 21:11:36 +01:00
parent 2d6473249b
commit 74833ba8a2
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mlflow_plugin_proxy_auth/proxy_auth_header_provider.py
View File

@ -1,3 +1,5 @@
"""Mlflow plugin proxy auth"""
import base64 import base64
import os import os
@ -8,6 +10,8 @@ from requests.auth import AuthBase
class ProxyAuthProvider(RequestAuthProvider): class ProxyAuthProvider(RequestAuthProvider):
""""Mlflow plugin class"""
def __init__(self): def __init__(self):
self.username = os.getenv("MLFLOW_PROXY_USERNAME") self.username = os.getenv("MLFLOW_PROXY_USERNAME")
self.password = os.getenv("MLFLOW_PROXY_PASSWORD") self.password = os.getenv("MLFLOW_PROXY_PASSWORD")
@ -20,6 +24,8 @@ class ProxyAuthProvider(RequestAuthProvider):
class ProxyAuth(AuthBase): class ProxyAuth(AuthBase):
"""Requests proxy auth class"""
def __init__(self, username, password): def __init__(self, username, password):
self.username = username self.username = username
self.password = password self.password = password