diff --git a/fmtr/tools/__init__.py b/fmtr/tools/__init__.py index 7a4c4dc..f24d803 100644 --- a/fmtr/tools/__init__.py +++ b/fmtr/tools/__init__.py @@ -107,6 +107,21 @@ except ImportError as exception: json_fix = MissingExtraMockModule('json_fix', exception) +try: + from fmtr.tools import semantic_tools as semantic +except ImportError as exception: + semantic = MissingExtraMockModule('semantic', exception) + +try: + from fmtr.tools import metric_tools as metric +except ImportError as exception: + metric = MissingExtraMockModule('metric', exception) + +try: + from fmtr.tools import html_tools as html +except ImportError as exception: + html = MissingExtraMockModule('html', exception) + __all__ = [ 'config', diff --git a/fmtr/tools/version b/fmtr/tools/version index b3ec163..2bd77c7 100644 --- a/fmtr/tools/version +++ b/fmtr/tools/version @@ -1 +1 @@ -0.9.3 \ No newline at end of file +0.9.4 \ No newline at end of file diff --git a/requirements.py b/requirements.py index 00d0fb7..1d0a822 100644 --- a/requirements.py +++ b/requirements.py @@ -23,7 +23,10 @@ 'api': ['fastapi', 'uvicorn', 'logging', 'dm'], 'ai': ['peft', 'transformers[sentencepiece]', 'torchvision', 'torchaudio', 'dm'], 'dm': ['pydantic'], - 'json-fix': ['json_repair'] + 'json-fix': ['json_repair'], + 'semantic': ['sentence_transformers', 'pandas'], + 'metric': ['pandas'], + 'html': ['html2text'], } CONSOLE_SCRIPTS = [