Hydra fails to load NFNZooDataset

[tufPenny]

I am trying to run the cnn experiments but they all fail when experiments/data_generalization.py tries to run line 17. And from what I can tell, vit_to_tg_data is never defined in any file in this repository. What am I missing?

Traceback (most recent call last):
  File "C:\Users\XXX\anaconda3\envs\neural-graphs\lib\site-packages\hydra\_internal\utils.py", line 650, in _locate
    obj = import_module(mod)
  File "C:\Users\XXX\anaconda3\envs\neural-graphs\lib\importlib\__init__.py", line 127, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
  File "<frozen importlib._bootstrap>", line 1030, in _gcd_import
  File "<frozen importlib._bootstrap>", line 1007, in _find_and_load
  File "<frozen importlib._bootstrap>", line 986, in _find_and_load_unlocked
  File "<frozen importlib._bootstrap>", line 680, in _load_unlocked
  File "<frozen importlib._bootstrap_external>", line 850, in exec_module
  File "<frozen importlib._bootstrap>", line 228, in _call_with_frames_removed
  File "C:\Users\XXX\repos\neural-graphs\experiments\data_generalization.py", line 17, in <module>
    from experiments.transformer_generalization.utils import vit_to_tg_data
ModuleNotFoundError: No module named 'experiments.transformer_generalization'

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "C:\Users\XXX\anaconda3\envs\neural-graphs\lib\site-packages\hydra\_internal\instantiate\_instantiate2.py", line 134, in _resolve_target
    target = _locate(target)
  File "C:\Users\XXX\anaconda3\envs\neural-graphs\lib\site-packages\hydra\_internal\utils.py", line 653, in _locate
    raise ImportError(
ImportError: Error loading 'experiments.data_generalization.NFNZooDataset':
ModuleNotFoundError("No module named 'experiments.transformer_generalization'")
Are you sure that 'data_generalization' is importable from module 'experiments'?

[IceIce1ce]

+1