akera commited on
Commit
c0c3e37
·
verified ·
1 Parent(s): f3c92ba

Update src/test_set.py

Browse files
Files changed (1) hide show
  1. src/test_set.py +1 -0
src/test_set.py CHANGED
@@ -5,6 +5,7 @@ from datasets import Dataset, load_dataset
5
  from typing import Dict, Tuple
6
  import salt.dataset
7
  from config import *
 
8
 
9
  def generate_test_set(max_samples_per_pair: int = MAX_TEST_SAMPLES) -> pd.DataFrame:
10
  """Generate standardized test set from SALT dataset."""
 
5
  from typing import Dict, Tuple
6
  import salt.dataset
7
  from config import *
8
+ from src.utils import get_all_language_pairs
9
 
10
  def generate_test_set(max_samples_per_pair: int = MAX_TEST_SAMPLES) -> pd.DataFrame:
11
  """Generate standardized test set from SALT dataset."""