Spaces:
Sleeping
Sleeping
minor fix
Browse files
qwen_classifier/evaluate.py
CHANGED
@@ -17,7 +17,9 @@ from .globals import global_model, global_tokenizer
|
|
17 |
|
18 |
eval_limit = os.getenv("EVAL_LIM")
|
19 |
if not eval_limit:
|
20 |
-
eval_limit
|
|
|
|
|
21 |
|
22 |
def _load_data(test_data_path):
|
23 |
test_data_path = Path(__file__).parent / test_data_path
|
|
|
17 |
|
18 |
eval_limit = os.getenv("EVAL_LIM")
|
19 |
if not eval_limit:
|
20 |
+
eval_limit = EVAL_LIMIT
|
21 |
+
else:
|
22 |
+
eval_limit = int(eval_limit)
|
23 |
|
24 |
def _load_data(test_data_path):
|
25 |
test_data_path = Path(__file__).parent / test_data_path
|