From 8de3cfd64e74c1e4832e585954cdbf0ee9676eb3 Mon Sep 17 00:00:00 2001 From: maleksandrov Date: Fri, 26 Mar 2021 15:55:25 +0100 Subject: [PATCH] updated package_data in eolearn.mask setup to use new model files --- mask/setup.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/mask/setup.py b/mask/setup.py index b2e669039..b1874e894 100644 --- a/mask/setup.py +++ b/mask/setup.py @@ -39,10 +39,12 @@ def get_version(): license='MIT', packages=find_packages(), include_package_data=True, - package_data={'eolearn.mask': [ - 'models/pixel_s2_cloud_detector_lightGBM_v0.2.joblib.dat', - 'models/ssim_s2_cloud_detector_lightGBM_v0.2.joblib.dat', - ]}, + package_data={ + 'eolearn.mask': [ + 'models/pixel_s2_cloud_detector_lightGBM_v0.2.txt', + 'models/ssim_s2_cloud_detector_lightGBM_v0.2.txt' + ] + }, install_requires=parse_requirements("requirements.txt"), zip_safe=False )