Skip to content

Commit 5e60c6a

Browse files
committed
.
1 parent 6d5b8e0 commit 5e60c6a

2 files changed

Lines changed: 3 additions & 25 deletions

File tree

deepeval/test_run/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,4 +12,4 @@
1212

1313
from .hooks import on_test_run_end, invoke_test_run_end_hook
1414
from .api import MetricData
15-
from .hyperparameters import log_hyperparameters, auto_log_hyperparameters
15+
from .hyperparameters import log_hyperparameters

deepeval/test_run/hyperparameters.py

Lines changed: 2 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
from typing import Union, Any, Dict
22

33
from deepeval.test_run import global_test_run_manager
4-
from deepeval.test_run.test_run import TEMP_FILE_PATH
54
from deepeval.prompt import Prompt, PromptApi
5+
from deepeval.test_run.test_run import TEMP_FILE_PATH
66

77

88
def process_hyperparameters(
@@ -62,26 +62,4 @@ def wrapper(*args, **kwargs):
6262
return func(*args, **kwargs)
6363

6464
# Return the wrapper function to be used as the decorator
65-
return wrapper
66-
67-
def auto_log_hyperparameters(hyperparameters: Dict[str, Any]) -> None:
68-
test_run = global_test_run_manager.get_test_run()
69-
if test_run is None:
70-
return
71-
existing_hyperparameters = getattr(test_run, "hyperparameters") or {}
72-
merged_hyperparameters: Dict[str, Any] = existing_hyperparameters.copy()
73-
for key, value in hyperparameters.items():
74-
if key not in merged_hyperparameters:
75-
merged_hyperparameters[key] = value
76-
else:
77-
if merged_hyperparameters[key] != value:
78-
suffix = 1
79-
new_key = f"{key}_{suffix}"
80-
while new_key in merged_hyperparameters:
81-
suffix += 1
82-
new_key = f"{key}_{suffix}"
83-
merged_hyperparameters[new_key] = value
84-
else:
85-
merged_hyperparameters[key] = value
86-
test_run.hyperparameters = process_hyperparameters(merged_hyperparameters)
87-
global_test_run_manager.save_test_run(TEMP_FILE_PATH)
65+
return wrapper

0 commit comments

Comments
 (0)