Skip to content

Commit

Permalink
fix(rank.py) remove the print code, sorry about that
Browse files Browse the repository at this point in the history
Signed-off-by: Marchons <[email protected]>
  • Loading branch information
Yoda-wu committed Oct 30, 2024
1 parent b118931 commit 468c5b5
Show file tree
Hide file tree
Showing 4 changed files with 0 additions and 9 deletions.
4 changes: 0 additions & 4 deletions core/storymanager/rank/rank.py
Original file line number Diff line number Diff line change
Expand Up @@ -138,7 +138,6 @@ def _sort_all_df(self, all_df, all_metric_names):

if metric_name not in all_metric_names:
continue
print(metric_name)
sort_metric_list.append(metric_name)
is_ascend_list.append(ele.get(metric_name) == "ascend")

Expand Down Expand Up @@ -234,15 +233,12 @@ def _draw_pictures(self, test_cases, test_results):
out_put = test_case.output_dir
test_result = test_results[test_case.id][0]
matrix = test_result.get("Matrix")
# print(out_put)
for key in matrix.keys():
draw_heatmap_picture(out_put, key, matrix[key])

def _prepare(self, test_cases, test_results, output_dir):
all_metric_names = self._get_all_metric_names(test_results)
print(f"in_prepare all_metric_names: {all_metric_names}")
all_hps_names = self._get_all_hps_names(test_cases)
print(f"in_prepare all_hps_names: {all_hps_names}")
all_module_types = self._get_all_module_types(test_cases)
self.all_df_header = [
"algorithm", *all_metric_names,
Expand Down
2 changes: 0 additions & 2 deletions core/testcasecontroller/metrics/metrics.py
Original file line number Diff line number Diff line change
Expand Up @@ -147,7 +147,6 @@ def forget_rate_func(system_metric_info: dict):
"""
info = system_metric_info.get(SystemMetricType.FORGET_RATE.value)
forget_rate = np.mean(info)
print(f"forget_rate: {forget_rate}")
return round(forget_rate, 3)


Expand All @@ -169,7 +168,6 @@ def get_metric_func(metric_dict: dict):

name = metric_dict.get("name")
url = metric_dict.get("url")
print(f"get metric func: name={name}, url={url}")
if url:
try:
load_module(url)
Expand Down
2 changes: 0 additions & 2 deletions core/testcasecontroller/testcase/testcase.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,6 @@ def run(self, workspace):
test_env_config = {}
# pylint: disable=C0103
for k, v in self.test_env.__dict__.items():
print(k,v)
test_env_config[k] = v

self.output_dir = self._get_output_dir(workspace)
Expand Down Expand Up @@ -112,7 +111,6 @@ def compute_metrics(self, paradigm_result, dataset, **kwargs):
metric_res = {}
system_metric_types = [e.value for e in SystemMetricType.__members__.values()]
for metric_name, metric_func in metric_funcs.items():
#print(metric_name)
if metric_name in system_metric_types:
metric_res[metric_name] = metric_func(kwargs)
else:
Expand Down
1 change: 0 additions & 1 deletion core/testenvmanager/dataset/dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -523,7 +523,6 @@ def load_data(

if data_format == DatasetFormat.TXT.value:
data = TxtDataParse(data_type=data_type, func=feature_process)
# print(file)
data.parse(file, use_raw=use_raw)

if data_format == DatasetFormat.JSON.value:
Expand Down

0 comments on commit 468c5b5

Please sign in to comment.