diff --git a/setup.py b/setup.py index 656b17c71..33c860477 100644 --- a/setup.py +++ b/setup.py @@ -126,7 +126,7 @@ setuptools.setup( "hcrystalball==0.1.10", "pytorch-forecasting>=0.9.0", ], - "benchmark": ["catboost>=0.26", "psutil==5.8.0", "xgboost==1.3.3"], + "benchmark": ["catboost>=0.26", "psutil==5.8.0", "xgboost==1.3.3", "pandas==1.1.4"], "openai": ["openai==0.27.4", "diskcache"], "autogen": ["openai==0.27.4", "diskcache", "docker"], "synapse": [