Spaces:
Sleeping
Sleeping
Update src/test_set.py
Browse files- src/test_set.py +2 -1
src/test_set.py
CHANGED
|
@@ -3,8 +3,9 @@ import pandas as pd
|
|
| 3 |
import yaml
|
| 4 |
from datasets import Dataset, load_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:
|
|
|
|
| 3 |
import yaml
|
| 4 |
from datasets import Dataset, load_dataset
|
| 5 |
from typing import Dict, Tuple
|
|
|
|
| 6 |
from config import *
|
| 7 |
+
import salt.dataset
|
| 8 |
+
|
| 9 |
from src.utils import get_all_language_pairs
|
| 10 |
|
| 11 |
def generate_test_set(max_samples_per_pair: int = MAX_TEST_SAMPLES) -> pd.DataFrame:
|