Skip to content

Commit ae1bf59

Browse files
authored
Merge pull request #20 from AI-Hypercomputer/shangkun-evaluation-cleanup
refactor: update model reference to refer to constant defined in hitl and adjust gitignore test file pattern
2 parents bc7af25 + 649d637 commit ae1bf59

6 files changed

Lines changed: 3 additions & 782 deletions

File tree

MaxKernel/evaluation/.gitignore

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
.pytest_cache/
22
.env
3-
*functional_test.py
3+
*_test.py
44
test_data/*

MaxKernel/evaluation/code_adapter/code_adapter.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,7 @@
88
adapt_reference_prompt,
99
)
1010
from evaluation.custom_types.kernel_task import KernelTask
11-
12-
GEMINI_MODEL = "gemini-2.5-flash"
11+
from hitl_agent.constants import MODEL_NAME
1312

1413
logging.basicConfig(
1514
level=logging.INFO,
@@ -60,7 +59,7 @@ def adapt(
6059
while attempt < self.max_retries:
6160
try:
6261
response = self.client.models.generate_content(
63-
model=GEMINI_MODEL, contents=prompt, config=config
62+
model=MODEL_NAME, contents=prompt, config=config
6463
)
6564
code = response.text.strip()
6665
if code.startswith("```python"):

MaxKernel/evaluation/code_adapter/code_adapter_test.py

Lines changed: 0 additions & 151 deletions
This file was deleted.

MaxKernel/evaluation/evaluation_utils_test.py

Lines changed: 0 additions & 189 deletions
This file was deleted.

0 commit comments

Comments
 (0)