Skip to content

Commit c3fc71d

Browse files
authored
chore: cleanup unncessary comments/casts to ignore type warnings (#1392)
1 parent d378d79 commit c3fc71d

File tree

13 files changed

+20
-23
lines changed

13 files changed

+20
-23
lines changed

examples/custom_output_files/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ def mutate(
9696

9797
@cocoindex.op.function()
9898
def markdown_to_html(text: str) -> str:
99-
return _markdown_it.render(text) # type: ignore
99+
return _markdown_it.render(text)
100100

101101

102102
@cocoindex.flow_def(name="CustomOutputFiles")

examples/face_recognition/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def extract_face_embedding(
8585
np.array(img),
8686
known_face_locations=[(0, img.width - 1, img.height - 1, 0)],
8787
)[0]
88-
return embedding # type: ignore
88+
return embedding
8989

9090

9191
@cocoindex.flow_def(name="FaceRecognition")

examples/fastapi_server_docker/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ async def lifespan(app: FastAPI) -> AsyncIterator[None]:
109109
fastapi_app = FastAPI(lifespan=lifespan)
110110

111111

112-
@fastapi_app.get("/search") # type: ignore
112+
@fastapi_app.get("/search")
113113
def search_endpoint(
114114
request: Request,
115115
q: str = Query(..., description="Search query"),

examples/image_search/colpali_main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ async def lifespan(app: FastAPI) -> AsyncIterator[None]:
100100

101101

102102
# --- Search API ---
103-
@app.get("/search") # type: ignore
103+
@app.get("/search")
104104
def search(
105105
q: str = Query(..., description="Search query"),
106106
limit: int = Query(5, description="Number of results"),

examples/image_search/main.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ def embed_query(text: str) -> list[float]:
3636
inputs = processor(text=[text], return_tensors="pt", padding=True)
3737
with torch.no_grad():
3838
features = model.get_text_features(**inputs)
39-
return features[0].tolist() # type: ignore
39+
return features[0].tolist()
4040

4141

4242
@cocoindex.op.function(cache=True, behavior_version=1, gpu=True)
@@ -51,7 +51,7 @@ def embed_image(
5151
inputs = processor(images=image, return_tensors="pt")
5252
with torch.no_grad():
5353
features = model.get_image_features(**inputs)
54-
return features[0].tolist() # type: ignore
54+
return features[0].tolist()
5555

5656

5757
# CocoIndex flow: Ingest images, extract captions, embed, export to Qdrant
@@ -140,7 +140,7 @@ async def lifespan(app: FastAPI) -> AsyncIterator[None]:
140140

141141

142142
# --- Search API ---
143-
@app.get("/search") # type: ignore
143+
@app.get("/search")
144144
def search(
145145
q: str = Query(..., description="Search query"),
146146
limit: int = Query(5, description="Number of results"),

examples/manuals_llm_extraction/main.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
from marker.models import create_model_dict
66
from marker.output import text_from_rendered
77
from marker.config.parser import ConfigParser
8-
from typing import cast
98

109
import cocoindex
1110

@@ -32,7 +31,7 @@ def __call__(self, content: bytes) -> str:
3231
temp_file.write(content)
3332
temp_file.flush()
3433
text_any, _, _ = text_from_rendered(self._converter(temp_file.name))
35-
return cast(str, text_any)
34+
return text_any
3635

3736

3837
@dataclasses.dataclass

examples/paper_metadata/main.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
from marker.models import create_model_dict
1010
from marker.output import text_from_rendered
1111
from functools import cache
12-
from typing import cast
1312
from pypdf import PdfReader, PdfWriter
1413

1514

@@ -68,7 +67,7 @@ def pdf_to_markdown(content: bytes) -> str:
6867
temp_file.write(content)
6968
temp_file.flush()
7069
text_any, _, _ = text_from_rendered(get_marker_converter()(temp_file.name))
71-
return cast(str, text_any)
70+
return text_any
7271

7372

7473
@cocoindex.flow_def(name="PaperMetadata")

examples/patient_intake_extraction_dspy/main.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ class Patient(BaseModel):
7777

7878

7979
# DSPy Signature for patient information extraction from images
80-
class PatientExtractionSignature(dspy.Signature): # type: ignore[misc]
80+
class PatientExtractionSignature(dspy.Signature):
8181
"""Extract structured patient information from a medical intake form image."""
8282

8383
form_images: list[dspy.Image] = dspy.InputField(
@@ -88,7 +88,7 @@ class PatientExtractionSignature(dspy.Signature): # type: ignore[misc]
8888
)
8989

9090

91-
class PatientExtractor(dspy.Module): # type: ignore[misc]
91+
class PatientExtractor(dspy.Module):
9292
"""DSPy module for extracting patient information from intake form images."""
9393

9494
def __init__(self) -> None:
@@ -98,7 +98,7 @@ def __init__(self) -> None:
9898
def forward(self, form_images: list[dspy.Image]) -> Patient:
9999
"""Extract patient information from form images and return as a Pydantic model."""
100100
result = self.extract(form_images=form_images)
101-
return result.patient # type: ignore
101+
return result.patient
102102

103103

104104
@cocoindex.op.function(cache=True, behavior_version=1)
@@ -123,7 +123,7 @@ def extract_patient(pdf_content: bytes) -> Patient:
123123
extractor = PatientExtractor()
124124
patient = extractor(form_images=form_images)
125125

126-
return patient # type: ignore
126+
return patient
127127

128128

129129
@cocoindex.flow_def(name="PatientIntakeExtractionDSPy")

examples/pdf_elements_embedding/main.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@
77
from dataclasses import dataclass
88
from pypdf import PdfReader
99
from transformers import CLIPModel, CLIPProcessor
10-
from typing import cast
1110

1211

1312
QDRANT_GRPC_URL = "http://localhost:6334"
@@ -36,7 +35,7 @@ def clip_embed_image(img_bytes: bytes) -> list[float]:
3635
inputs = processor(images=image, return_tensors="pt")
3736
with torch.no_grad():
3837
features = model.get_image_features(**inputs)
39-
return cast(list[float], features[0].tolist())
38+
return features[0].tolist()
4039

4140

4241
def clip_embed_query(text: str) -> list[float]:
@@ -47,7 +46,7 @@ def clip_embed_query(text: str) -> list[float]:
4746
inputs = processor(text=[text], return_tensors="pt", padding=True)
4847
with torch.no_grad():
4948
features = model.get_text_features(**inputs)
50-
return cast(list[float], features[0].tolist())
49+
return features[0].tolist()
5150

5251

5352
@cocoindex.transform_flow()

examples/pdf_embedding/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def __call__(self, content: bytes) -> str:
3434
temp_file.write(content)
3535
temp_file.flush()
3636
text_any, _, _ = text_from_rendered(self._converter(temp_file.name))
37-
return text_any # type: ignore
37+
return text_any
3838

3939

4040
@cocoindex.transform_flow()

0 commit comments

Comments
 (0)