fix the pydantic 'protected_namespaces' warning

om-ai-lab · Jul 25, 2024 · 3d25208 · 3d25208
1 parent daabfeb
commit 3d25208
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/engine/tools/ovd_tool.py b/engine/tools/ovd_tool.py
@@ -33,6 +33,10 @@ class ObjectDetection(BaseModelTool):
                         "Tasks like object counting, specific object detection, etc. must use this tool.")
     ovd_endpoint: str
     model_id: str = 'OmDet-Turbo_tiny_SWIN_T'
+
+    class Config:
+        """Configuration for this pydantic object."""
+        protected_namespaces = ()
 
     def _run(
             self, timestamps: str, labels: str