-
Notifications
You must be signed in to change notification settings - Fork 262
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
feat: sqlalchemy tests #622
Conversation
service = service_subclass() | ||
table_tuples = service.get_all_entries() | ||
self.assertTrue( | ||
len(table_tuples) < 100 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just curious, is it generally bad to have a large catalog table?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Not really, I just added an assert for checking the contents of the table for now.
eva/udfs/udf_bootstrap_queries.py
Outdated
Open_udf_query, | ||
Similarity_udf_query | ||
# Disabled because required packages (eg., easy_ocr might not be preinstalled) | ||
# YoloV5_udf_query, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
We shouldn't disable this for release
mode as else early users will have to manually run the create udf query. We can condition on mode here.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done. We now have three modes: debug/release: all UDFS, minimal: No YOLO
@@ -25,5 +25,8 @@ def setup_pytorch_tests(): | |||
CatalogManager().reset() | |||
execute_query_fetch_all("LOAD VIDEO 'data/ua_detrac/ua_detrac.mp4' INTO MyVideo;") | |||
execute_query_fetch_all("LOAD VIDEO 'data/mnist/mnist.mp4' INTO MNIST;") | |||
load_inbuilt_udfs() | |||
load_udfs_for_testing() | |||
from eva.udfs.udf_bootstrap_queries import YoloV5_udf_query |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Any reason for having this separately?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
In case we want a more complex logic for picking UDFs later during testing
YOLO
is particular test cases (to avoid slowing down all test cases)