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

View File

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