2024-03-04 11:10:52 +08:00
|
|
|
from mmengine.config import read_base
|
|
|
|
|
|
|
|
with read_base():
|
|
|
|
from .needlebench_4k.needlebench import needlebench_datasets as needlebench_datasets_4k
|
|
|
|
from .needlebench_8k.needlebench import needlebench_datasets as needlebench_datasets_8k
|
|
|
|
from .needlebench_32k.needlebench import needlebench_datasets as needlebench_datasets_32k
|
|
|
|
from .needlebench_128k.needlebench import needlebench_datasets as needlebench_datasets_128k
|
|
|
|
from .needlebench_200k.needlebench import needlebench_datasets as needlebench_datasets_200k
|
2024-02-28 21:40:00 +08:00
|
|
|
from .needlebench_1000k.needlebench import needlebench_datasets as needlebench_datasets_1000k
|
2024-03-04 11:10:52 +08:00
|
|
|
|
|
|
|
needlebench_datasets = sum((v for k, v in locals().items() if k.endswith('_datasets')), [])
|