Skip to content

Commit 8de3cfd

Browse files
committed
updated package_data in eolearn.mask setup to use new model files
1 parent cb709fb commit 8de3cfd

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

mask/setup.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -39,10 +39,12 @@ def get_version():
3939
license='MIT',
4040
packages=find_packages(),
4141
include_package_data=True,
42-
package_data={'eolearn.mask': [
43-
'models/pixel_s2_cloud_detector_lightGBM_v0.2.joblib.dat',
44-
'models/ssim_s2_cloud_detector_lightGBM_v0.2.joblib.dat',
45-
]},
42+
package_data={
43+
'eolearn.mask': [
44+
'models/pixel_s2_cloud_detector_lightGBM_v0.2.txt',
45+
'models/ssim_s2_cloud_detector_lightGBM_v0.2.txt'
46+
]
47+
},
4648
install_requires=parse_requirements("requirements.txt"),
4749
zip_safe=False
4850
)

0 commit comments

Comments
 (0)