Update unit tests to use the new BaseModel types

This commit is contained in:
Saba
2021-12-05 09:31:39 -05:00
parent c4cd4b57f1
commit 76e9e9da2f
5 changed files with 19 additions and 16 deletions

View File

@@ -27,7 +27,7 @@ class TextSearchConfigModel(ConfigBaseModel):
class ImageSearchConfigModel(ConfigBaseModel):
use_xmp_metadata: Optional[str]
batch_size: Optional[int]
input_directory: Optional[List[str]]
input_directory: Optional[Path]
input_filter: Optional[str]
embeddings_file: Optional[Path]