diff --git a/olah/utils/logging.py b/olah/utils/logging.py index dac6bc0..86cb6e4 100644 --- a/olah/utils/logging.py +++ b/olah/utils/logging.py @@ -15,10 +15,6 @@ import sys from typing import AsyncGenerator, Generator import warnings - -import requests -import torch - from olah.constants import DEFAULT_LOGGER_DIR handler = None diff --git a/pyproject.toml b/pyproject.toml index 2adb5ce..a17661d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -13,9 +13,9 @@ classifiers = [ "License :: OSI Approved :: MIT License", ] dependencies = [ - "fastapi", "fastapi-utils", "httpx", "numpy", "pydantic<=2.8.2", "requests", "toml", + "fastapi", "fastapi-utils", "httpx", "numpy", "pydantic<=2.8.2", "pydantic-settings<=2.4.0", "requests", "toml", "rich>=10.0.0", "shortuuid", "uvicorn", "tenacity>=8.2.2", "pytz", "cachetools", "GitPython", - "PyYAML" + "PyYAML", "typing_inspect>=0.9.0", "huggingface_hub", "jinja2" ] [project.optional-dependencies] diff --git a/requirements.txt b/requirements.txt index f1ec968..404bbcd 100644 --- a/requirements.txt +++ b/requirements.txt @@ -11,3 +11,5 @@ cachetools==5.4.0 PyYAML==6.0.1 tenacity==8.5.0 peewee==3.17.6 +typing_inspect==0.9.0 +jinja2==3.1.4