fixes #60: modify request_function to use endpoint_name

ai-cfia · Apr 15, 2024 · c99eccc · c99eccc
1 parent 94823a5
commit c99eccc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/model/__init__.py b/model/__init__.py
@@ -4,8 +4,8 @@
 from model.six_seeds import request_inference_from_nachet_6seeds
 
 request_function = {
-    "swin_endpoint": request_inference_from_swin,
-    "seed_detector": request_inference_from_seed_detector,
+    "swin-endpoint": request_inference_from_swin,
+    "seed-detector": request_inference_from_seed_detector,
     "test": request_inference_from_test,
-    "nachet_6seeds": request_inference_from_nachet_6seeds
+    "nachet-6seeds": request_inference_from_nachet_6seeds
 }
diff --git a/pipelines/pipeline_template.yaml b/pipelines/pipeline_template.yaml
@@ -5,7 +5,6 @@ date:
 
 pipelines:
   - models:
-    model_name:
     pipeline_name:
     created_by:
     creation_date: