From 030eeec7211fdfd96765c7a277d82459b4b77354 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Beno=C3=AEt=20Mal=C3=A9zieux?= Date: Wed, 14 Aug 2024 09:44:50 +0200 Subject: [PATCH] FIX: change model type name for landmark detection to be coherent with what's inside detectors --- giskard_vision/landmark_detection/models/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/giskard_vision/landmark_detection/models/base.py b/giskard_vision/landmark_detection/models/base.py index 738427e3..95047c20 100644 --- a/giskard_vision/landmark_detection/models/base.py +++ b/giskard_vision/landmark_detection/models/base.py @@ -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: