updated package_data in eolearn.mask setup to use new model files

sentinel-hub · Mar 26, 2021 · 8de3cfd · 8de3cfd
1 parent cb709fb
commit 8de3cfd
Showing 1 changed file with 6 additions and 4 deletions.
diff --git 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
 )