Merge pull request #523 from vprashar2929/fix-docker-cred

fix(ci): update the credentails in release workflow

Author: Sunil Thaha

.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()