@@ -49,70 +49,70 @@ type FVTPredictor struct {
4949
5050// Array of all the predictors that need to be tested
5151var predictorsArray = []FVTPredictor {
52- {
53- predictorName : "tf" ,
54- predictorFilename : "tf-predictor.yaml" ,
55- currentModelPath : "fvt/tensorflow/mnist.savedmodel" ,
56- updatedModelPath : "fvt/tensorflow/mnist-dup.savedmodel" ,
57- differentPredictorName : "onnx" ,
58- differentPredictorFilename : "onnx-predictor.yaml" ,
59- },
60- {
61- predictorName : "keras" ,
62- predictorFilename : "keras-predictor.yaml" ,
63- currentModelPath : "fvt/tensorflow/keras-mnist/mnist.h5" ,
64- updatedModelPath : "fvt/tensorflow/keras-mnistnew/mnist.h5" ,
65- differentPredictorName : "tf" ,
66- differentPredictorFilename : "tf-predictor.yaml" ,
67- },
68- {
69- predictorName : "onnx" ,
70- predictorFilename : "onnx-predictor.yaml" ,
71- currentModelPath : "fvt/onnx/onnx-mnist" ,
72- updatedModelPath : "fvt/onnx/onnx-mnist-new" ,
73- differentPredictorName : "pytorch" ,
74- differentPredictorFilename : "pytorch-predictor.yaml" ,
75- },
76- {
77- predictorName : "onnx-withschema" ,
78- predictorFilename : "onnx-predictor-withschema.yaml" ,
79- currentModelPath : "fvt/onnx/onnx-withschema" ,
80- updatedModelPath : "fvt/onnx/onnx-withschema-new" ,
81- differentPredictorName : "pytorch" ,
82- differentPredictorFilename : "pytorch-predictor.yaml" ,
83- },
84- {
85- predictorName : "pytorch" ,
86- predictorFilename : "pytorch-predictor.yaml" ,
87- currentModelPath : "fvt/pytorch/pytorch-cifar" ,
88- updatedModelPath : "fvt/pytorch/pytorch-cifar-new" ,
89- differentPredictorName : "onnx" ,
90- differentPredictorFilename : "onnx-predictor.yaml" ,
91- },
92- {
93- predictorName : "xgboost" ,
94- predictorFilename : "xgboost-predictor.yaml" ,
95- currentModelPath : "fvt/xgboost/mushroom" ,
96- updatedModelPath : "fvt/xgboost/mushroom-dup" ,
97- differentPredictorName : "onnx" ,
98- differentPredictorFilename : "onnx-predictor.yaml" ,
99- },
100- {
101- predictorName : "lightgbm" ,
102- predictorFilename : "lightgbm-predictor.yaml" ,
103- currentModelPath : "fvt/lightgbm/mushroom" ,
104- updatedModelPath : "fvt/lightgbm/mushroom-dup" ,
105- differentPredictorName : "onnx" ,
106- differentPredictorFilename : "onnx-predictor.yaml" ,
107- },
108- {
109- predictorName : "openvino" ,
110- predictorFilename : "openvino-mnist-predictor.yaml" ,
111- currentModelPath : "fvt/openvino/mnist" ,
112- updatedModelPath : "fvt/openvino/mnist-dup" ,
113- differentPredictorName : "xgboost" ,
114- differentPredictorFilename : "xgboost-predictor.yaml" ,
115- },
52+ // {
53+ // predictorName: "tf",
54+ // predictorFilename: "tf-predictor.yaml",
55+ // currentModelPath: "fvt/tensorflow/mnist.savedmodel",
56+ // updatedModelPath: "fvt/tensorflow/mnist-dup.savedmodel",
57+ // differentPredictorName: "onnx",
58+ // differentPredictorFilename: "onnx-predictor.yaml",
59+ // },
60+ // {
61+ // predictorName: "keras",
62+ // predictorFilename: "keras-predictor.yaml",
63+ // currentModelPath: "fvt/tensorflow/keras-mnist/mnist.h5",
64+ // updatedModelPath: "fvt/tensorflow/keras-mnistnew/mnist.h5",
65+ // differentPredictorName: "tf",
66+ // differentPredictorFilename: "tf-predictor.yaml",
67+ // },
68+ // {
69+ // predictorName: "onnx",
70+ // predictorFilename: "onnx-predictor.yaml",
71+ // currentModelPath: "fvt/onnx/onnx-mnist",
72+ // updatedModelPath: "fvt/onnx/onnx-mnist-new",
73+ // differentPredictorName: "pytorch",
74+ // differentPredictorFilename: "pytorch-predictor.yaml",
75+ // },
76+ // {
77+ // predictorName: "onnx-withschema",
78+ // predictorFilename: "onnx-predictor-withschema.yaml",
79+ // currentModelPath: "fvt/onnx/onnx-withschema",
80+ // updatedModelPath: "fvt/onnx/onnx-withschema-new",
81+ // differentPredictorName: "pytorch",
82+ // differentPredictorFilename: "pytorch-predictor.yaml",
83+ // },
84+ // {
85+ // predictorName: "pytorch",
86+ // predictorFilename: "pytorch-predictor.yaml",
87+ // currentModelPath: "fvt/pytorch/pytorch-cifar",
88+ // updatedModelPath: "fvt/pytorch/pytorch-cifar-new",
89+ // differentPredictorName: "onnx",
90+ // differentPredictorFilename: "onnx-predictor.yaml",
91+ // },
92+ // {
93+ // predictorName: "xgboost",
94+ // predictorFilename: "xgboost-predictor.yaml",
95+ // currentModelPath: "fvt/xgboost/mushroom",
96+ // updatedModelPath: "fvt/xgboost/mushroom-dup",
97+ // differentPredictorName: "onnx",
98+ // differentPredictorFilename: "onnx-predictor.yaml",
99+ // },
100+ // {
101+ // predictorName: "lightgbm",
102+ // predictorFilename: "lightgbm-predictor.yaml",
103+ // currentModelPath: "fvt/lightgbm/mushroom",
104+ // updatedModelPath: "fvt/lightgbm/mushroom-dup",
105+ // differentPredictorName: "onnx",
106+ // differentPredictorFilename: "onnx-predictor.yaml",
107+ // },
108+ // {
109+ // predictorName: "openvino",
110+ // predictorFilename: "openvino-mnist-predictor.yaml",
111+ // currentModelPath: "fvt/openvino/mnist",
112+ // updatedModelPath: "fvt/openvino/mnist-dup",
113+ // differentPredictorName: "xgboost",
114+ // differentPredictorFilename: "xgboost-predictor.yaml",
115+ // },
116116 {
117117 predictorName : "xgboost-fil" ,
118118 predictorFilename : "xgboost-fil-predictor.yaml" ,
@@ -121,14 +121,14 @@ var predictorsArray = []FVTPredictor{
121121 differentPredictorName : "onnx" ,
122122 differentPredictorFilename : "onnx-predictor.yaml" ,
123123 },
124- {
125- predictorName : "lightgbm-fil" ,
126- predictorFilename : "lightgbm-fil-predictor.yaml" ,
127- currentModelPath : "fvt/lightgbm/mushroom-fil" ,
128- updatedModelPath : "fvt/lightgbm/mushroom-fil-dup" ,
129- differentPredictorName : "onnx" ,
130- differentPredictorFilename : "onnx-predictor.yaml" ,
131- },
124+ // {
125+ // predictorName: "lightgbm-fil",
126+ // predictorFilename: "lightgbm-fil-predictor.yaml",
127+ // currentModelPath: "fvt/lightgbm/mushroom-fil",
128+ // updatedModelPath: "fvt/lightgbm/mushroom-fil-dup",
129+ // differentPredictorName: "onnx",
130+ // differentPredictorFilename: "onnx-predictor.yaml",
131+ // },
132132 // TorchServe test is currently disabled
133133 // {
134134 // predictorName: "pytorch-mar",
@@ -1137,7 +1137,7 @@ var _ = Describe("Inference service", Ordered, func() {
11371137 var _ = Describe ("test " + i .name + " isvc" , Ordered , func () {
11381138 var isvcName string
11391139
1140- FIt ("should successfully load a model" , func () {
1140+ It ("should successfully load a model" , func () {
11411141 isvcObject := NewIsvcForFVT (i .inferenceServiceFileName )
11421142 isvcName = isvcObject .GetName ()
11431143 CreateIsvcAndWaitAndExpectReady (isvcObject , PredictorTimeout )
@@ -1150,9 +1150,9 @@ var _ = Describe("Inference service", Ordered, func() {
11501150 ExpectSuccessfulInference_sklearnMnistSvm (isvcName )
11511151 })
11521152
1153- // AfterAll(func() {
1154- // FVTClientInstance.DeleteIsvc(isvcName)
1155- // })
1153+ AfterAll (func () {
1154+ FVTClientInstance .DeleteIsvc (isvcName )
1155+ })
11561156
11571157 })
11581158 }
0 commit comments