Changed: Fix bento dependencies (#527)

* Changed: move pydantic deps from extras to regular dependencies * Changed: use <= in bento build sdk dependency
konfuzio-ai · Aug 7, 2024 · 4cc8846 · 4cc8846
1 parent 3f8e4bd
commit 4cc8846
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 3 deletions.
diff --git a/extras_list.py b/extras_list.py
@@ -23,7 +23,6 @@
         'chardet==5.1.0',
         'datasets==2.14.6',
         'evaluate==0.4.1',
-        'pydantic>2,<2.8',
         'spacy>=2.3.5',
         'torch>=1.8.1',
         'torchvision>=0.9.1',

diff --git a/konfuzio_sdk/settings_importer.py b/konfuzio_sdk/settings_importer.py
@@ -41,7 +41,6 @@ def is_dependency_installed(dependency: str) -> bool:
 
 extras = [
     'chardet==5.1.0',
-    'pydantic>2,<2.8',
     'torch>=1.8',
     'torchvision>=0.9',
     'transformers>=4.21.2',

diff --git a/konfuzio_sdk/trainer/information_extraction.py b/konfuzio_sdk/trainer/information_extraction.py
@@ -853,7 +853,7 @@ def build_bento(self, bento_model):
                 ],
                 labels=self.bento_metadata,
                 python={
-                    'packages': [f'konfuzio-sdk=={self.konfuzio_sdk_version}'],
+                    'packages': [f'konfuzio-sdk<={self.konfuzio_sdk_version}'],
                     'lock_packages': True,
                 },
                 build_ctx=temp_dir,

diff --git a/setup.py b/setup.py
@@ -93,6 +93,7 @@
         'numpy==1.23.5',
         'pandas>=1.3.5,<2.0.0',
         'Pillow>=8.4.0',
+        'pydantic>2,<2.8',
         'python-dateutil>=2.8.2',
         'python-decouple>=3.3',
         'python-dotenv>=1.0,<1.1',