Spaces:
Running
Running
Update src/test_set.py
Browse files- 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."""
|