|
9 | 9 | import importlib_resources |
10 | 10 | import pandas as pd |
11 | 11 | import requests |
12 | | -from SUITPy import ( |
13 | | - fetch_buckner_2011, |
14 | | - fetch_diedrichsen_2009, |
15 | | - fetch_ji_2019, |
16 | | - fetch_king_2019, |
17 | | - fetch_xue_2021, |
18 | | -) |
| 12 | +from SUITPy.atlas import fetch_atlas |
19 | 13 | from SUITPy.utils import _fetch_files, _get_dataset_dir |
20 | 14 |
|
21 | 15 | from dmriseg._utils import fill_doc |
@@ -220,15 +214,15 @@ def fetch_suit_atlas(atlas, **kwargs): |
220 | 214 | _data = kwargs.pop(data_arg, _data) |
221 | 215 |
|
222 | 216 | if atlas == Atlas.BUCKNER: |
223 | | - atlas_data = fetch_buckner_2011(**kwargs) |
| 217 | + atlas_data = fetch_atlas("Buckner_2011", **kwargs) |
224 | 218 | elif atlas == Atlas.DIEDRICHSEN: |
225 | | - atlas_data = fetch_diedrichsen_2009(**kwargs) |
| 219 | + atlas_data = fetch_atlas("Diedrichsen_2009", **kwargs) |
226 | 220 | elif atlas == Atlas.JI: |
227 | | - atlas_data = fetch_ji_2019(**kwargs) |
| 221 | + atlas_data = fetch_atlas("Ji_2019", **kwargs) |
228 | 222 | elif atlas == Atlas.KING: |
229 | | - atlas_data = fetch_king_2019(data=_data, **kwargs) |
| 223 | + atlas_data = fetch_atlas("King_2019", data=_data, **kwargs) |
230 | 224 | elif atlas == Atlas.XUE: |
231 | | - atlas_data = fetch_xue_2021(**kwargs) |
| 225 | + atlas_data = fetch_atlas("Xue_2021", **kwargs) |
232 | 226 | else: |
233 | 227 | raise ValueError(f"Unknown atlas name: {atlas}.") |
234 | 228 |
|
@@ -285,7 +279,7 @@ def fetch_suit_cmap_lut_files( |
285 | 279 | # name matching the URL |
286 | 280 | dataset_name = atlas_dirname.value.lower() |
287 | 281 | data_dir = _get_dataset_dir( |
288 | | - dataset_name, data_dir=data_dir, verbose=verbose |
| 282 | + dataset_name, atlas_dir=data_dir, verbose=verbose |
289 | 283 | ) |
290 | 284 |
|
291 | 285 | # Get local fullpath(s) of downloaded file(s) |
|
0 commit comments