Skip to content

Commit faa844d

Browse files
committed
Fix last mypy issues
1 parent 2c20cad commit faa844d

File tree

3 files changed

+3
-5
lines changed

3 files changed

+3
-5
lines changed

.github/workflows/test.yml

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,5 +34,3 @@ jobs:
3434
run: "python -m unittest"
3535
- name: Check types with mypy
3636
run: mypy .
37-
# TODO: Fix typing
38-
continue-on-error: true

roboflow/models/semantic_segmentation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def __init__(self, api_key: str, version_id: str):
1818
super(SemanticSegmentationModel, self).__init__(api_key, version_id)
1919
self.api_url = f"{SEMANTIC_SEGMENTATION_URL}/{self.dataset_id}/{self.version}"
2020

21-
def predict(self, image_path: str, confidence: int = 50):
21+
def predict(self, image_path: str, confidence: int = 50): # type: ignore[override]
2222
"""
2323
Infers detections based on image from a specified model and image path.
2424

roboflow/models/video.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def __init__(
5656
""" # noqa: E501 // docs
5757
self.__api_key = api_key
5858

59-
def predict(
59+
def predict( # type: ignore[override]
6060
self,
6161
video_path: str,
6262
inference_type: str,
@@ -124,7 +124,7 @@ def predict(
124124
{
125125
"model_id": self.dataset_id,
126126
"model_version": self.version,
127-
"inference_type": self.inference_type,
127+
"inference_type": inference_type,
128128
}
129129
]
130130

0 commit comments

Comments
 (0)