1  // This proto file defines internal interfaces for MLflow, e.g
 2  // enums used for storage in Tracking or Model Registry
 3  
 4  syntax = "proto2";
 5  
 6  package mlflow.internal;
 7  
 8  import "scalapb/scalapb.proto";
 9  
10  option java_package = "org.mlflow.internal.proto";
11  option py_generic_services = true;
12  option (scalapb.options) = {flat_package: true};
13  
14  // Types of vertices represented in MLflow Run Inputs. Valid vertices are MLflow objects that can
15  // have an input relationship.
16  enum InputVertexType {
17    RUN = 1;
18  
19    DATASET = 2;
20  
21    MODEL = 3;
22  }
23  
24  // Types of vertices represented in MLflow Run Outputs. Valid vertices are MLflow objects that can
25  // have an output relationship.
26  enum OutputVertexType {
27    RUN_OUTPUT = 1;
28  
29    MODEL_OUTPUT = 2;
30  }