From 74833ba8a2d1f740b52eef803855a67a3040e9b4 Mon Sep 17 00:00:00 2001 From: LordMathis Date: Sun, 17 Dec 2023 21:11:36 +0100 Subject: [PATCH] Add docstrings --- mlflow_plugin_proxy_auth/proxy_auth_header_provider.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/mlflow_plugin_proxy_auth/proxy_auth_header_provider.py b/mlflow_plugin_proxy_auth/proxy_auth_header_provider.py index dd8e453..437b139 100644 --- a/mlflow_plugin_proxy_auth/proxy_auth_header_provider.py +++ b/mlflow_plugin_proxy_auth/proxy_auth_header_provider.py @@ -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