nbaldwin commited on
Commit
005876e
·
1 Parent(s): 5abf8a7

changed dependencies to main

Browse files
Files changed (1) hide show
  1. run.py +2 -9
run.py CHANGED
@@ -31,7 +31,6 @@ flow_verse.sync_dependencies(dependencies)
31
  if __name__ == "__main__":
32
 
33
  #1. ~~~~~ Set up a colink server ~~~~
34
- FLOW_MODULES_PATH = "./"
35
 
36
  cl = start_colink_server()
37
 
@@ -63,7 +62,7 @@ if __name__ == "__main__":
63
  )
64
 
65
  #4. ~~~~~Start A Worker Thread~~~~~
66
- run_dispatch_worker_thread(cl, dispatch_point="coflows_dispatch", flow_modules_base_path=FLOW_MODULES_PATH)
67
 
68
  #5. ~~~~~Mount the flow and get its proxy~~~~~~
69
  proxy_flow= serve_utils.get_flow_instance(
@@ -77,13 +76,7 @@ if __name__ == "__main__":
77
  data = {"id": 0}
78
 
79
 
80
- #option1: use the FlowMessage class
81
- input_message = FlowMessage(
82
- data=data,
83
- )
84
-
85
- #option2: use the proxy_flow
86
- #input_message = proxy_flow.package_input_message(data = data)
87
 
88
  #7. ~~~ Run inference ~~~
89
  future = proxy_flow.get_reply_future(input_message)
 
31
  if __name__ == "__main__":
32
 
33
  #1. ~~~~~ Set up a colink server ~~~~
 
34
 
35
  cl = start_colink_server()
36
 
 
62
  )
63
 
64
  #4. ~~~~~Start A Worker Thread~~~~~
65
+ run_dispatch_worker_thread(cl)
66
 
67
  #5. ~~~~~Mount the flow and get its proxy~~~~~~
68
  proxy_flow= serve_utils.get_flow_instance(
 
76
  data = {"id": 0}
77
 
78
 
79
+ input_message = proxy_flow.package_input_message(data = data)
 
 
 
 
 
 
80
 
81
  #7. ~~~ Run inference ~~~
82
  future = proxy_flow.get_reply_future(input_message)