diff --git a/brainscore_vision/models/alexnet_7be5be79/setup.py b/brainscore_vision/models/alexnet_7be5be79/setup.py index 421914cfb..41c4ef930 100644 --- a/brainscore_vision/models/alexnet_7be5be79/setup.py +++ b/brainscore_vision/models/alexnet_7be5be79/setup.py @@ -3,9 +3,10 @@ from setuptools import setup, find_packages -requirements = [ "torchvision", - "torch" -] +requirements = ["torchvision", + "torch", + "fire" + ] setup( packages=find_packages(exclude=['tests']),