fix(ci): update the credentails in release workflow

This commit updates the credentials in the release workflow to use the
correct secrets defined for the repository.

This also updates the extractor and profiler code to make linting and
formatting CI pass

Signed-off-by: vprashar2929 <[email protected]>

Author: vprashar2929

.github/workflows/release.yml

@@ -21,8 +21,8 @@ jobs:
         uses: docker/login-action@v2
         with:
           registry: ${{ vars.IMAGE_REGISTRY }}
-          username: ${{ secrets.docker_username }}
-          password: ${{ secrets.docker_password }}
+          username: ${{ secrets.BOT_NAME }}
+          password: ${{ secrets.BOT_TOKEN }}
 
       - name: Git set user
         shell: bash

src/kepler_model/train/extractor/extractor.py

@@ -228,7 +228,7 @@ def get_workload_feature_data(self, query_results, features, use_vm_metrics=Fals
         if len(feature_to_remove) != 0:
             features = self.process_feature(features, feature_to_remove, cur_accelerator_features)
         # return with reset index for later aggregation
-        #print(feature_data.reset_index().to_string())
+        # print(feature_data.reset_index().to_string())
         return feature_data.reset_index(), features
 
     def get_system_feature_data(self, query_results, features):
@@ -262,8 +262,8 @@ def get_power_data(self, query_results, energy_components, source, use_vm_metric
                 aggr_query_data = aggr_query_data.loc[aggr_query_data["job"] != VM_JOB_NAME]
             # filter source
             aggr_query_data = aggr_query_data[aggr_query_data[SOURCE_COL] == source]
-            #print("aggr query data power")
-            #print(aggr_query_data.to_string())
+            # print("aggr query data power")
+            # print(aggr_query_data.to_string())
             if len(aggr_query_data) == 0:
                 return None
             if unit_col is not None:
@@ -311,7 +311,7 @@ def get_power_data(self, query_results, energy_components, source, use_vm_metric
         if len(power_data_list) == 0:
             return None
         power_data = pd.concat(power_data_list, axis=1).dropna()
-        #print(power_data.to_string())
+        # print(power_data.to_string())
         return power_data
 
     def get_system_category(self, query_results):

src/kepler_model/train/profiler/node_type_index.py

@@ -266,9 +266,14 @@ def get_node_type(self, in_spec: NodeTypeSpec, loose_search: bool = False):
             return -1, -1, -1
         compare_spec = in_spec.copy()
         num_of_none = get_num_of_none(compare_spec)
-        similarity_map, max_similarity, most_similar_index, has_candidate, candidate_uncertain_attribute_freq, candidate_uncertain_attribute_total = (
-            self._find_candidates(in_spec, loose_search)
-        )
+        (
+            similarity_map,
+            max_similarity,
+            most_similar_index,
+            has_candidate,
+            candidate_uncertain_attribute_freq,
+            candidate_uncertain_attribute_total,
+        ) = self._find_candidates(in_spec, loose_search)
         if max_similarity == 1:
             return most_similar_index, 0, 0
         if has_candidate:

tests/estimator_model_request_test.py

@@ -126,9 +126,9 @@ def test_model_request():
     power_request = json.loads(data, object_hook=lambda d: PowerRequest(**d))
     output_path = get_achived_model(power_request)
     assert output_path is None, f"model should be invalid\n {output_path}"
-    os.environ["MODEL_CONFIG"] = (
-        f"{estimator_enable_key}=true\n{init_url_key}={get_url(energy_source=energy_source, output_type=output_type, feature_group=FeatureGroup.BPFOnly, model_topurl=model_topurl, pipeline_name=default_train_output_pipeline)}\n"
-    )
+    os.environ[
+        "MODEL_CONFIG"
+    ] = f"{estimator_enable_key}=true\n{init_url_key}={get_url(energy_source=energy_source, output_type=output_type, feature_group=FeatureGroup.BPFOnly, model_topurl=model_topurl, pipeline_name=default_train_output_pipeline)}\n"
     set_env_from_model_config()
     print("Requesting from ", os.environ[init_url_key])
     reset_failed_list()