Skip to content

Commit 7bed7ee

Browse files
committed
Fix service collection
* Use the correct graph to collect service
1 parent 7e04e5b commit 7bed7ee

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

aidb/engine/engine.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -66,8 +66,9 @@ async def clear_ml_cache(self, service_name_list = None):
6666
# Get all the services that need to be cleared because of foreign key constraints
6767
for bounded_service in service_ordering:
6868
if bounded_service.service.name in service_name_list:
69-
for in_edge in self._config.table_graph.in_edges(bounded_service.service.name):
70-
service_name_list.add(in_edge[0])
69+
for input_column in bounded_service.binding.input_columns:
70+
for in_edge in self._config.inference_graph.in_edges(input_column):
71+
service_name_list.add(in_edge.bound_service.service.name)
7172

7273
# Clear the services in reversed topological order
7374
for bounded_service in reversed(service_ordering):

0 commit comments

Comments
 (0)