Updated setup.py with XGBoost

setup.py

@@ -9,7 +9,7 @@ setuptools.setup(
     description="A package to predict PROTAC-induced protein degradation.",
     long_description=open("README.md").read(),
     packages=setuptools.find_packages(),
-    install_requires=["torch", "pytorch_lightning", "scikit-learn", "imbalanced-learn", "rdkit", "pandas", "joblib", "h5py", "optuna", "torchmetrics"],
+    install_requires=["torch", "pytorch_lightning", "scikit-learn", "imbalanced-learn", "rdkit", "pandas", "joblib", "h5py", "optuna", "torchmetrics", "xgboost"],
     classifiers=[
         "Programming Language :: Python :: 3",
         "Programming Language :: Python :: 3.6",
@@ -18,4 +18,4 @@ setuptools.setup(
     ],
     include_package_data=True,
     package_data={"": ["data/*.h5", "data/*.pkl", "data/*.csv", "models/*.ckpt"]},
-)
+)