nbaldwin commited on
Commit
fdafd1b
1 Parent(s): dc701b1
Files changed (1) hide show
  1. run.py +4 -4
run.py CHANGED
@@ -10,7 +10,7 @@ from aiflows.utils.general_helpers import read_yaml_file, quick_load_api_keys
10
  from aiflows import logging
11
  from aiflows.flow_cache import CACHING_PARAMETERS, clear_cache
12
 
13
- from aiflows.utils import serve_utils
14
  from aiflows.workers import run_dispatch_worker_thread
15
  from aiflows.messages import FlowMessage
16
  from aiflows.interfaces import KeyInterface
@@ -52,7 +52,7 @@ if __name__ == "__main__":
52
 
53
 
54
  #3. ~~~~ Serve The Flow ~~~~
55
- serve_utils.serve_flow(
56
  cl = cl,
57
  flow_class_name="flow_modules.aiflows.VisionFlowModule.VisionAtomicFlow",
58
  flow_endpoint="VisionAtomicFlow",
@@ -62,7 +62,7 @@ if __name__ == "__main__":
62
  run_dispatch_worker_thread(cl)
63
 
64
  #5. ~~~~~Mount the flow and get an instance of it via a proxy~~~~~~
65
- proxy_flow= serve_utils.get_flow_instance(
66
  cl=cl,
67
  flow_endpoint="VisionAtomicFlow",
68
  user_id="local",
@@ -118,5 +118,5 @@ if __name__ == "__main__":
118
 
119
 
120
  #9. ~~~~~Optional: Unserve Flow~~~~~~
121
- # serve_utils.delete_served_flow(cl, "VisionFlowModule")
122
 
 
10
  from aiflows import logging
11
  from aiflows.flow_cache import CACHING_PARAMETERS, clear_cache
12
 
13
+ from aiflows.utils import serving
14
  from aiflows.workers import run_dispatch_worker_thread
15
  from aiflows.messages import FlowMessage
16
  from aiflows.interfaces import KeyInterface
 
52
 
53
 
54
  #3. ~~~~ Serve The Flow ~~~~
55
+ serving.serve_flow(
56
  cl = cl,
57
  flow_class_name="flow_modules.aiflows.VisionFlowModule.VisionAtomicFlow",
58
  flow_endpoint="VisionAtomicFlow",
 
62
  run_dispatch_worker_thread(cl)
63
 
64
  #5. ~~~~~Mount the flow and get an instance of it via a proxy~~~~~~
65
+ proxy_flow= serving.get_flow_instance(
66
  cl=cl,
67
  flow_endpoint="VisionAtomicFlow",
68
  user_id="local",
 
118
 
119
 
120
  #9. ~~~~~Optional: Unserve Flow~~~~~~
121
+ # serving.delete_served_flow(cl, "VisionFlowModule")
122