Merge pull request #1 from Nader-gator/main

add pyworker v2
2024-09-04 11:53:45 -07:00
parent 7cd1a30393 589216d15f
commit ed7a952bf4
31 changed files with 3000 additions and 1 deletions
@@ -0,0 +1,7 @@
+from lib.test_utils import test_load_cmd, test_args
+from .data_types import InputData
+
+WORKER_ENDPOINT = "/generate"
+
+if __name__ == "__main__":
+    test_load_cmd(InputData, WORKER_ENDPOINT, arg_parser=test_args)