Skip to content

Commit

Permalink
FIX: change model type name for landmark detection to be coherent wit…
Browse files Browse the repository at this point in the history
…h what's inside detectors
  • Loading branch information
bmalezieux committed Aug 14, 2024
1 parent 0af2c9b commit 030eeec
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion giskard_vision/landmark_detection/models/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
class FaceLandmarksModelBase(ModelBase):
"""Abstract class that serves as a template for all landmark model predictions"""

model_type = "landmark_detection"
model_type = "landmark"
prediction_result_cls = Types.prediction_result

def __init__(self, n_landmarks: int, n_dimensions: int, name: Optional[str] = None) -> None:
Expand Down

0 comments on commit 030eeec

Please sign in to comment.