We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7e04e5b commit 7bed7eeCopy full SHA for 7bed7ee
aidb/engine/engine.py
@@ -66,8 +66,9 @@ async def clear_ml_cache(self, service_name_list = None):
66
# Get all the services that need to be cleared because of foreign key constraints
67
for bounded_service in service_ordering:
68
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])
+ for input_column in bounded_service.binding.input_columns:
+ for in_edge in self._config.inference_graph.in_edges(input_column):
71
+ service_name_list.add(in_edge.bound_service.service.name)
72
73
# Clear the services in reversed topological order
74
for bounded_service in reversed(service_ordering):
0 commit comments