[fix] add missing models in tests

edenai · Dec 20, 2024 · c6bb767 · c6bb767
1 parent 8f03cd0
commit c6bb767
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/edenai_apis/features/image/object_detection/object_detection_args.py b/edenai_apis/features/image/object_detection/object_detection_args.py
@@ -7,8 +7,6 @@
 from edenai_apis.utils.files import FileInfo, FileWrapper
 
 
-
-
 def object_detection_arguments(provider_name: str) -> Dict:
     feature_path = os.path.dirname(os.path.dirname(__file__))
 
@@ -25,4 +23,4 @@ def object_detection_arguments(provider_name: str) -> Dict:
         mediainfo(image_path).get("channels", "1"),
     )
     file_wrapper = FileWrapper(image_path, "", file_info)
-    return {"file": file_wrapper, "settings": {}}
+    return {"file": file_wrapper, "settings": {"clarifai": "general-image-detection"}}
diff --git a/edenai_apis/features/text/summarize/summarize_args.py b/edenai_apis/features/text/summarize/summarize_args.py
@@ -4,5 +4,9 @@ def summarize_arguments(provider_name: str):
         "output_sentences": 3,
         "text": "Barack Hussein Obama is an American politician who served as the 44th president of the United States from 2009 to 2017. A member of the Democratic Party, Obama was the first African-American president of the United States. He previously served as a U.S. senator from Illinois from 2005 to 2008 and as an Illinois state senator from 1997 to 2004.",
         "language": "en",
-        "settings": {"openai": "gpt-4", "nlpcloud": "finetuned-llama-2-70b"},
+        "settings": {
+            "openai": "gpt-4",
+            "nlpcloud": "finetuned-llama-2-70b",
+            "anthropic": "claude-v2",
+        },
     }