He entrenado un modelo LightGBM y ahora quiero crear una rutina de predicción personalizada para implementar en google aiplatform.

Al cargar el archivo del modelo, arroja un error de que lightgbm no está instalado. Usé el siguiente código en setup.py pero no funciona.

 from setuptools import setup, find_packages DEPENDENCIES = ['lightgbm==3.1.1'] setup( name='my_custom_code', version='0.4', install_requires=DEPENDENCIES, include_package_date=True, packages=find_packages(), description='custom code.', scripts=['predictor.py', 'preprocess.py'])