KeivanR commited on
Commit
a511d9b
·
1 Parent(s): 1c3c329
Files changed (1) hide show
  1. qwen_classifier/evaluate.py +3 -1
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 == 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