chore: hatch fmt

Signed-off-by: Sunil Thaha <[email protected]>

Author: Sunil Thaha

src/kepler_model/estimate/archived_model.py

@@ -1,4 +1,5 @@
 import logging
+
 import requests
 from requests_file import FileAdapter
 
@@ -45,7 +46,7 @@ def is_valid_model(metrics, metadata, filters):
 
     for attrb, val in filters.items():
         if not hasattr(metadata, attrb) or getattr(metadata, attrb) is None:
-            logger.warn(f"{metadata['model_name']} has no {attrb}")
+            logger.warning(f"{metadata['model_name']} has no {attrb}")
             return False
 
         cmp_val = getattr(metadata, attrb)
@@ -73,7 +74,7 @@ def get_achived_model(power_request):
     output_type = ModelOutputType[power_request.output_type]
     url = get_init_model_url(power_request.energy_source, output_type_name)
     if url == "":
-        logger.warn("no URL set for ", output_type_name, power_request.energy_source)
+        logger.warning(f"no URL set for {output_type_name}, {power_request.energy_source}")
         return None
     logger.info(f"try getting archieved model from URL: {url} for {output_type_name}")
 
@@ -94,7 +95,7 @@ def get_achived_model(power_request):
                 failed_list += [output_type_name]
                 return None
         except Exception as e:
-            logger.warn(f"cannot validate the archived model: {e}")
+            logger.warning(f"cannot validate the archived model: {e}")
             return None
 
     return output_path

src/kepler_model/estimate/estimator.py

@@ -13,7 +13,7 @@
 from kepler_model.estimate.model.model import load_downloaded_model
 from kepler_model.estimate.model_server_connector import is_model_server_enabled, make_request
 from kepler_model.train.profiler.node_type_index import NodeTypeSpec, discover_spec_values, get_machine_spec
-from kepler_model.util.config import CONFIG_PATH, SERVE_SOCKET, download_path, set_env_from_model_config, set_config_dir
+from kepler_model.util.config import CONFIG_PATH, SERVE_SOCKET, download_path, set_config_dir, set_env_from_model_config
 from kepler_model.util.loader import get_download_output_path, load_metadata
 from kepler_model.util.train_types import ModelOutputType, convert_enery_source, is_output_type_supported
 

src/kepler_model/util/config.py

@@ -12,8 +12,8 @@
 #
 #################################################
 
-import os
 import logging
+import os
 
 import requests