Spaces:
Sleeping
Sleeping
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."""
|