From 4fc3276501c939af71c2eda70c124034189a5389 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Wed, 15 Feb 2023 16:32:53 +0100
Subject: [PATCH] fix lint
---
dan/mlflow.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/dan/mlflow.py b/dan/mlflow.py
index 366a119a..180406f3 100644
--- a/dan/mlflow.py
+++ b/dan/mlflow.py
@@ -3,9 +3,8 @@ import os
from contextlib import contextmanager
import mlflow
-from mlflow.environment_variables import MLFLOW_HTTP_REQUEST_MAX_RETRIES
-
import requests
+from mlflow.environment_variables import MLFLOW_HTTP_REQUEST_MAX_RETRIES
from dan import logger
@@ -20,6 +19,7 @@ def make_mlflow_request(mlflow_method, *args, **kwargs):
logger.error(f"Call to `{str(mlflow_method)}` failed with error: {str(e)}")
raise e
+
def setup_environment(config: dict):
"""
Get the necessary variables from the config file and put them in the environment variables
@@ -39,7 +39,9 @@ def setup_environment(config: dict):
# Check max retry setting
max_retries = MLFLOW_HTTP_REQUEST_MAX_RETRIES.get()
if max_retries and int(max_retries) <= 1:
- logger.warning(f"The maximum number of retries for MLflow HTTP requests is set to {max_retries}, which is low. Consider using a higher value.")
+ logger.warning(
+ f"The maximum number of retries for MLflow HTTP requests is set to {max_retries}, which is low. Consider using a higher value."
+ )
def logging_metrics(
--
GitLab