ModuleNotFoundError: No module named 'torch._six'

[rafaelblevin821]

ModuleNotFoundError: No module named 'torch._six'

[Hananel-Hazan]

I'm unable to replicate this error, can you guide me how to replicate?

[rafaelblevin821]

I'm unable to replicate this error, can you guide me how to replicate?

@Hananel-Hazan to replicate this error, you would need to install a fresh install of bindsnet using: pip install git+https://github.com/BindsNET/bindsnet.git and run pytest...

(bindsnet) [********@login1 [kelvin2] bindsnet]$ python3 -m pytest test
==================================================================== test session starts =====================================================================
platform linux -- Python 3.10.11, pytest-7.3.2, pluggy-1.2.0
rootdir: /users/********/bindsnet
collected 12 items / 3 errors

=========================================================================== ERRORS ===========================================================================
______________________________________________________ ERROR collecting test/analysis/test_analyzers.py ______________________________________________________
ImportError while importing test module '/users/********/bindsnet/test/analysis/test_analyzers.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
/mnt/scratch2/users/********/conda/envs/bindsnet/lib/python3.10/importlib/__init__.py:126: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
test/analysis/test_analyzers.py:6: in <module>
    from bindsnet.analysis.pipeline_analysis import MatplotlibAnalyzer, TensorboardAnalyzer
bindsnet/__init__.py:3: in <module>
    from bindsnet import (
bindsnet/datasets/__init__.py:2: in <module>
    from bindsnet.datasets.collate import time_aware_collate
bindsnet/datasets/collate.py:13: in <module>
    from torch._six import string_classes
E   ModuleNotFoundError: No module named 'torch._six'
______________________________________________________ ERROR collecting test/encoding/test_encoding.py _______________________________________________________
ImportError while importing test module '/users/dharkin/bindsnet/test/encoding/test_encoding.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
/mnt/scratch2/users/********/conda/envs/bindsnet/lib/python3.10/importlib/__init__.py:126: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
test/encoding/test_encoding.py:3: in <module>
    from bindsnet.encoding import *
bindsnet/__init__.py:3: in <module>
    from bindsnet import (
bindsnet/datasets/__init__.py:2: in <module>
    from bindsnet.datasets.collate import time_aware_collate
bindsnet/datasets/collate.py:13: in <module>
    from torch._six import string_classes
E   ModuleNotFoundError: No module named 'torch._six'
________________________________________________________ ERROR collecting test/models/test_models.py _________________________________________________________
ImportError while importing test module '/users/dharkin/bindsnet/test/models/test_models.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
/mnt/scratch2/users/********/conda/envs/bindsnet/lib/python3.10/importlib/__init__.py:126: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
test/models/test_models.py:1: in <module>
    from bindsnet.models import DiehlAndCook2015, TwoLayerNetwork
bindsnet/__init__.py:3: in <module>
    from bindsnet import (
bindsnet/datasets/__init__.py:2: in <module>
    from bindsnet.datasets.collate import time_aware_collate
bindsnet/datasets/collate.py:13: in <module>
    from torch._six import string_classes
E   ModuleNotFoundError: No module named 'torch._six'
================================================================== short test summary info ===================================================================
ERROR test/analysis/test_analyzers.py
ERROR test/encoding/test_encoding.py
ERROR test/models/test_models.py
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! Interrupted: 3 errors during collection !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
================================================================ 3 errors in 72.48s (0:01:12) ================================================================

[Hananel-Hazan]

Thanks for elaborating.
I suspect that the installed BindsNET (3.1.2) and the version saved on the disk (running the tests) are not the same versions.
Can you please check this.

If you want to use the version on your disk, install it with pip install -e .

[rafaelblevin821]

Hi @Hananel-Hazan, I think the issue was on my end. I was running pytest on the wrong installation folder. It now seems to be working fine.