Skip to content

Commit

Permalink
#150 provided collection name placeholder and collection-name for inf…
Browse files Browse the repository at this point in the history
…erence script
  • Loading branch information
nicolay-r committed Jul 13, 2024
1 parent d2a9db1 commit 876f27d
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion arelight/pipelines/items/backend_d3js_graphs.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,7 @@ def iter_column_value(self, samples, column_value):

def apply_core(self, input_data, pipeline_ctx):

collection_name = pipeline_ctx.provide("d3js_collection_name")
predict_filepath = pipeline_ctx.provide("predict_filepath")
result_reader = pipeline_ctx.provide("predict_reader")
labels_fmt = pipeline_ctx.provide("labels_formatter")
Expand All @@ -81,7 +82,7 @@ def apply_core(self, input_data, pipeline_ctx):
labels = list(map(lambda item: labels_fmt.label_to_str(labels_scaler.uint_to_label(item)), uint_labels_iter))

graph = make_graph_from_relations_array(
graph_name="UNKNOWN_GRAPH_NAME",
graph_name=collection_name,
relations=self.__iter_relations(samples=samples,
labels=labels,
labels_filter_func=self.__graph_label_filter,
Expand Down

0 comments on commit 876f27d

Please sign in to comment.