brain-score / model-tools

Helper functions to extract model activations and translate from Machine Learning to Neuroscience
MIT License
8 stars 27 forks source link

add legacy support for LogitsBehavior class #63

Closed mschrimpf closed 2 years ago

mschrimpf commented 2 years ago

E.g. http://braintree.mit.edu:8080/job/run_benchmarks/2469/console is currently failing with

Traceback (most recent call last):
  File "/rdma/vast-rdma/scratch/Fri/score_models_env_2469/brain-score/brainscore/submission/repository.py", line 67, in install_project
    return import_module(package)
  File "/om2/group/dicarlo/jenkins/miniconda3/envs/score_models_2469/lib/python3.7/importlib/__init__.py", line 127, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
  File "<frozen importlib._bootstrap>", line 1006, in _gcd_import
  File "<frozen importlib._bootstrap>", line 983, in _find_and_load
  File "<frozen importlib._bootstrap>", line 967, in _find_and_load_unlocked
  File "<frozen importlib._bootstrap>", line 677, in _load_unlocked
  File "<frozen importlib._bootstrap_external>", line 728, in exec_module
  File "<frozen importlib._bootstrap>", line 219, in _call_with_frames_removed
  File "/om2/scratch/Fri/score_models_env_2469/work_dir/candidate_models/models/base_models.py", line 1, in <module>
    from candidate_models.base_models import base_model_pool
  File "/om2/scratch/Fri/score_models_env_2469/work_dir/candidate_models/candidate_models/__init__.py", line 4, in <module>
    from candidate_models.model_commitments import brain_translated_pool
  File "/om2/scratch/Fri/score_models_env_2469/work_dir/candidate_models/candidate_models/model_commitments/__init__.py", line 3, in <module>
    from candidate_models.model_commitments.cornets import cornet_brain_pool
  File "/om2/scratch/Fri/score_models_env_2469/work_dir/candidate_models/candidate_models/model_commitments/cornets.py", line 13, in <module>
    from model_tools.brain_transformation.behavior import BehaviorArbiter, LogitsBehavior, ProbabilitiesMapping
ImportError: cannot import name 'LogitsBehavior' from 'model_tools.brain_transformation.behavior' (/om2/scratch/Fri/score_models_env_2469/.local/lib/python3.7/site-packages/model_tools/brain_transformation/behavior.py)

This PR adds legacy support for the LogitsBehavior class which should fix that error.