diff --git a/tests/aliby/network/extraction/test_base.py b/tests/aliby/network/extraction/test_base.py index 3d806933fe9faada450220b8344da97729e1e796..9a6566a8a89071e7ad018fe4076581cc2cef1b50 100644 --- a/tests/aliby/network/extraction/test_base.py +++ b/tests/aliby/network/extraction/test_base.py @@ -7,7 +7,13 @@ from extraction.core.functions.loaders import ( load_funs, load_redfuns, ) -from extraction.examples import data +from extraction import local_data_loaders as data +from importlib_resources import files + +# FUTURETODO can be replaced by importlib.resources.files('aliby') after upgrading to 3.11 +files_dir = files("aliby") +data_dir = files_dir.parent.parent / "tests" / "agora" / "data" + dsets1z = data.load_1z() dsets = data.load()