Upgrade experimental package dependencies and use Poetry 1.6.1. (#11339)

Part of upgrading our CI to use Poetry 1.6.1.
This commit is contained in:
Predrag Gruevski
2023-10-16 21:13:31 -04:00
committed by GitHub
parent c2c0814a94
commit 7c0f1bf23f
9 changed files with 1621 additions and 1389 deletions

View File

@@ -39,7 +39,9 @@ class RELLM(HuggingFacePipeline):
default=200, description="Maximum number of new tokens to generate."
)
@root_validator
# TODO: move away from `root_validator` since it is deprecated in pydantic v2
# and causes mypy type-checking failures (hence the `type: ignore`)
@root_validator # type: ignore[call-overload]
def check_rellm_installation(cls, values: dict) -> dict:
import_rellm()
return values