mirror of
https://github.com/Aider-AI/aider.git
synced 2025-06-02 18:54:59 +00:00
style: Fix linting issues in models.py
This commit is contained in:
parent
74e60e98b7
commit
207a631a65
1 changed files with 6 additions and 6 deletions
|
@ -791,7 +791,7 @@ def sanity_check_model(io, model):
|
||||||
def check_for_dependencies(io, model_name):
|
def check_for_dependencies(io, model_name):
|
||||||
"""
|
"""
|
||||||
Check for model-specific dependencies and install them if needed.
|
Check for model-specific dependencies and install them if needed.
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
io: The IO object for user interaction
|
io: The IO object for user interaction
|
||||||
model_name: The name of the model to check dependencies for
|
model_name: The name of the model to check dependencies for
|
||||||
|
@ -803,14 +803,14 @@ def check_for_dependencies(io, model_name):
|
||||||
check_pip_install_extra(
|
check_pip_install_extra(
|
||||||
io, "boto3", "AWS Bedrock models require the boto3 package.", ["boto3"]
|
io, "boto3", "AWS Bedrock models require the boto3 package.", ["boto3"]
|
||||||
)
|
)
|
||||||
|
|
||||||
# Check if this is a Vertex AI model and ensure google-cloud-aiplatform is installed
|
# Check if this is a Vertex AI model and ensure google-cloud-aiplatform is installed
|
||||||
elif model_name.startswith("vertex_ai/"):
|
elif model_name.startswith("vertex_ai/"):
|
||||||
check_pip_install_extra(
|
check_pip_install_extra(
|
||||||
io,
|
io,
|
||||||
"google.cloud.aiplatform",
|
"google.cloud.aiplatform",
|
||||||
"Google Vertex AI models require the google-cloud-aiplatform package.",
|
"Google Vertex AI models require the google-cloud-aiplatform package.",
|
||||||
["google-cloud-aiplatform"]
|
["google-cloud-aiplatform"],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue