Skip to content

Commit

Permalink
create genai MCD preprocessor component (#2294)
Browse files Browse the repository at this point in the history
Co-authored-by: Richard Li <[email protected]>
  • Loading branch information
RichardLi1437 and Richard Li authored Feb 12, 2024
1 parent cdd0653 commit e31af43
Show file tree
Hide file tree
Showing 3 changed files with 59 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
type: component
spec: spec.yaml
categories: ["Model Monitoring"]
Original file line number Diff line number Diff line change
@@ -0,0 +1,50 @@
$schema: http://azureml/sdk-2-0/SparkComponent.json
type: spark

name: genai_mdc_preprocessor
display_name: GenAI MDC - Preprocessor
description: Filters the raw span log based on the window provided, and aggregates it to trace level.
version: 0.0.1
is_deterministic: true

code: ../../src
entry:
file: ./model_data_collector_preprocessor/genai_run.py

inputs:
data_window_start:
type: string
data_window_end:
type: string
input_data:
type: uri_folder
mode: direct
outputs:
preprocessed_span_data:
type: mltable
mode: direct
aggregated_trace_data:
type: mltable
mode: direct
conf:
spark.driver.cores: 4
spark.driver.memory: 28g
spark.executor.cores: 4
spark.executor.memory: 28g
spark.dynamicAllocation.enabled: True
spark.dynamicAllocation.minExecutors: 1
spark.dynamicAllocation.maxExecutors: 4
spark.synapse.library.python.env: |
channels:
- conda-forge
dependencies:
- python=3.8
- pip:
- mltable~=1.3.0
name: momo-base-spark
args: >-
--data_window_end ${{inputs.data_window_end}}
--data_window_start ${{inputs.data_window_start}}
--input_data ${{inputs.input_data}}
--preprocessed_span_data ${{outputs.preprocessed_span_data}}
--aggregated_trace_data ${{outputs.aggregated_trace_data}}
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.

"""Entry script for GenAI MDC Preprocessor."""

# place holder file, will add content later

0 comments on commit e31af43

Please sign in to comment.