From 0d6e844996b1b179ae27e98f70f96db8500b8d3f Mon Sep 17 00:00:00 2001 From: yuming <305248291@qq.com> Date: Mon, 6 Nov 2023 10:28:33 -0500 Subject: [PATCH 1/4] remove logger --- CHANGELOG.md | 3 ++- unstructured_inference/__version__.py | 2 +- unstructured_inference/models/base.py | 7 ------- 3 files changed, 3 insertions(+), 9 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 811f1f1c..e4488dde 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,5 +1,6 @@ -## 0.7.11-dev2 +## 0.7.11-dev3 +* fix: remove logger info for chipper * fix: update broken slack invite link in chipper logger info * enhancement: Improve error message when # images extracted doesn't match # page layouts. * fix: use automatic mixed precision on GPU for Chipper diff --git a/unstructured_inference/__version__.py b/unstructured_inference/__version__.py index cab69df5..1e6ab706 100644 --- a/unstructured_inference/__version__.py +++ b/unstructured_inference/__version__.py @@ -1 +1 @@ -__version__ = "0.7.11-dev2" # pragma: no cover +__version__ = "0.7.11-dev3" # pragma: no cover diff --git a/unstructured_inference/models/base.py b/unstructured_inference/models/base.py index 6e09a7a1..d8e3121b 100644 --- a/unstructured_inference/models/base.py +++ b/unstructured_inference/models/base.py @@ -54,13 +54,6 @@ def get_model(model_name: Optional[str] = None, **kwargs) -> UnstructuredModel: model = UnstructuredYoloXModel() initialize_params = {**YOLOX_MODEL_TYPES[model_name], **kwargs} elif model_name in CHIPPER_MODEL_TYPES: - logger.warning( - "The Chipper model is currently in Beta and is not yet ready for production use. " - "You can reach out to the Unstructured engineering team in the Unstructured " - "community Slack if you have any feedback on the Chipper model. " - "You can join the community Slack here: " - "https://short.unstructured.io/pzw05l7", - ) model = UnstructuredChipperModel() initialize_params = {**CHIPPER_MODEL_TYPES[model_name], **kwargs} elif model_name == "super_gradients": From d61d35bf3df708a45aef6d26243188064ecb83aa Mon Sep 17 00:00:00 2001 From: yuming <305248291@qq.com> Date: Mon, 6 Nov 2023 10:29:35 -0500 Subject: [PATCH 2/4] changelog nit --- CHANGELOG.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index e4488dde..1aa45702 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,6 +1,6 @@ ## 0.7.11-dev3 -* fix: remove logger info for chipper +* chore: remove logger info for chipper since its private * fix: update broken slack invite link in chipper logger info * enhancement: Improve error message when # images extracted doesn't match # page layouts. * fix: use automatic mixed precision on GPU for Chipper From 5550b02cc55e8b6b79ea9d19cb252c0afe1de592 Mon Sep 17 00:00:00 2001 From: yuming <305248291@qq.com> Date: Mon, 6 Nov 2023 10:46:07 -0500 Subject: [PATCH 3/4] lintlint --- unstructured_inference/models/base.py | 1 - 1 file changed, 1 deletion(-) diff --git a/unstructured_inference/models/base.py b/unstructured_inference/models/base.py index d8e3121b..53915635 100644 --- a/unstructured_inference/models/base.py +++ b/unstructured_inference/models/base.py @@ -1,6 +1,5 @@ from typing import Dict, Optional -from unstructured_inference.logger import logger from unstructured_inference.models.chipper import MODEL_TYPES as CHIPPER_MODEL_TYPES from unstructured_inference.models.chipper import UnstructuredChipperModel from unstructured_inference.models.detectron2 import ( From 34892229ef6cff5766250eaf2b66d5d8122c6a62 Mon Sep 17 00:00:00 2001 From: yuming <305248291@qq.com> Date: Mon, 6 Nov 2023 11:11:08 -0500 Subject: [PATCH 4/4] also remove the logger test for chipper --- test_unstructured_inference/models/test_model.py | 11 ----------- 1 file changed, 11 deletions(-) diff --git a/test_unstructured_inference/models/test_model.py b/test_unstructured_inference/models/test_model.py index ab3b3600..5839641b 100644 --- a/test_unstructured_inference/models/test_model.py +++ b/test_unstructured_inference/models/test_model.py @@ -35,17 +35,6 @@ def test_get_model(monkeypatch): assert isinstance(models.get_model("checkbox"), MockModel) -def test_get_model_warns_on_chipper(monkeypatch, caplog): - monkeypatch.setattr( - models, - "UnstructuredChipperModel", - MockModel, - ) - with mock.patch.object(models, "models", {}): - models.get_model("chipper") - assert caplog.records[0].levelname == "WARNING" - - def test_raises_invalid_model(): with pytest.raises(models.UnknownModelException): models.get_model("fake_model")