fix: make query and database required

vra · Jul 1, 2023 · 714c481 · 714c481
1 parent 35cd55f
commit 714c481
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/dinov2_retrieval/__init__.py b/src/dinov2_retrieval/__init__.py
@@ -31,12 +31,14 @@ def main():
         "-q",
         "--query",
         type=str,
+        required=True,
         help="path to a query image file or image folder",
     )
     parser.add_argument(
         "-d",
         "--database",
         type=str,
+        required=True,
         help="path to the database image file or image folder",
     )
     parser.add_argument(