Skip to content

Commit 1d17ebf

Browse files
committed
chore(typing): remove obsolete suppressions
Signed-off-by: Pouyanpi <13303554+Pouyanpi@users.noreply.github.com>
1 parent 07ecfe7 commit 1d17ebf

7 files changed

Lines changed: 12 additions & 12 deletions

File tree

nemoguardrails/cli/providers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def select_provider_type() -> Optional[ProviderType]:
8585

8686
# exact match only
8787
if result in provider_types:
88-
return cast(ProviderType, result) # type: ignore
88+
return cast(ProviderType, result)
8989

9090
# fuzzy match
9191
matches = [t for t in provider_types if result.lower() in t.lower()]

nemoguardrails/embeddings/cache.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
try:
2626
import redis # type: ignore
2727
except ImportError:
28-
redis = None # type: ignore
28+
redis = None
2929

3030
from nemoguardrails.rails.llm.config import EmbeddingsCacheConfig
3131

nemoguardrails/embeddings/providers/azureopenai.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ class AzureEmbeddingModel(EmbeddingModel):
4646

4747
def __init__(self, embedding_model: str):
4848
try:
49-
from openai import AzureOpenAI # type: ignore
49+
from openai import AzureOpenAI
5050
except ImportError:
5151
raise ImportError("Could not import openai, please install it with `pip install openai`.")
5252
# Set Azure OpenAI API credentials

nemoguardrails/embeddings/providers/fastembed.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ class FastEmbedEmbeddingModel(EmbeddingModel):
4242
engine_name = "FastEmbed"
4343

4444
def __init__(self, embedding_model: str, **kwargs):
45-
from fastembed import TextEmbedding as Embedding # type: ignore
45+
from fastembed import TextEmbedding as Embedding
4646

4747
# Enabling a short form model name for all-MiniLM-L6-v2.
4848
if embedding_model == "all-MiniLM-L6-v2":

nemoguardrails/embeddings/providers/nim.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ class NIMEmbeddingModel(EmbeddingModel):
3535

3636
def __init__(self, embedding_model: str, **kwargs):
3737
try:
38-
from langchain_nvidia_ai_endpoints import NVIDIAEmbeddings # type: ignore
38+
from langchain_nvidia_ai_endpoints import NVIDIAEmbeddings
3939

4040
self.model = embedding_model
4141
self.document_embedder = NVIDIAEmbeddings(model=embedding_model, **kwargs)

nemoguardrails/embeddings/providers/openai.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -46,11 +46,11 @@ def __init__(
4646
**kwargs,
4747
):
4848
try:
49-
import openai # type: ignore
50-
from openai import OpenAI # type: ignore
49+
import openai
50+
from openai import OpenAI
5151
except ImportError:
5252
raise ImportError("Could not import openai, please install it with `pip install openai`.")
53-
if openai.__version__ < "1.0.0": # type: ignore
53+
if openai.__version__ < "1.0.0":
5454
raise RuntimeError(
5555
"`openai<1.0.0` is no longer supported. Please upgrade using `pip install openai>=1.0.0`."
5656
)

nemoguardrails/tracing/adapters/opentelemetry.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -60,8 +60,8 @@
6060
if TYPE_CHECKING:
6161
from nemoguardrails.tracing import InteractionLog
6262
try:
63-
from opentelemetry import trace # type: ignore
64-
from opentelemetry.trace import NoOpTracerProvider # type: ignore
63+
from opentelemetry import trace
64+
from opentelemetry.trace import NoOpTracerProvider
6565

6666
except ImportError:
6767
raise ImportError(
@@ -100,7 +100,7 @@ def __init__(
100100
"""
101101

102102
# validate that OpenTelemetry is properly configured
103-
provider = trace.get_tracer_provider() # type: ignore
103+
provider = trace.get_tracer_provider()
104104
if provider is None or isinstance(provider, NoOpTracerProvider):
105105
warnings.warn(
106106
"No OpenTelemetry TracerProvider configured. Traces will not be exported. "
@@ -111,7 +111,7 @@ def __init__(
111111
stacklevel=2,
112112
)
113113

114-
self.tracer = trace.get_tracer( # type: ignore
114+
self.tracer = trace.get_tracer(
115115
service_name,
116116
instrumenting_library_version=version("nemoguardrails"),
117117
schema_url="https://opentelemetry.io/schemas/1.26.0",

0 commit comments

Comments
 (0)