Просмотр исходного кода

turn off dcgm

Signed-off-by: Sean Holcomb <seanholcomb@gmail.com>
Sean Holcomb 6 дней назад
Родитель
Сommit
7855d3bfa4

+ 3 - 1
core/pkg/compute/kubemodel/dcgmdevice_test.go

@@ -161,7 +161,9 @@ func TestComputeDCGMDevices(t *testing.T) {
 			km, err := NewKubeModel(testClusterUID, false, ds)
 			km, err := NewKubeModel(testClusterUID, false, ds)
 			require.NoError(t, err)
 			require.NoError(t, err)
 
 
-			kms, err := km.ComputeKubeModelSet(start, end)
+			kms := kubemodel.NewKubeModelSet(start, end)
+
+			err = km.computeDCGMDevices(kms, start, end)
 			require.NoError(t, err)
 			require.NoError(t, err)
 
 
 			assert.Equal(t, tt.want, kms.DCGMDevices)
 			assert.Equal(t, tt.want, kms.DCGMDevices)

+ 1 - 1
core/pkg/compute/kubemodel/kubemodel.go

@@ -104,6 +104,6 @@ func (km *KubeModel) computeFuncs(start, end time.Time) []computeFunc {
 		km.computePersistentVolumeClaims,
 		km.computePersistentVolumeClaims,
 		km.computePods,
 		km.computePods,
 		km.computeContainers,
 		km.computeContainers,
-		km.computeDCGMDevices,
+		//km.computeDCGMDevices,
 	}
 	}
 }
 }

+ 1 - 1
core/pkg/compute/kubemodel/kubemodel_test.go

@@ -185,7 +185,7 @@ func TestComputeKubeModelSet(t *testing.T) {
 				assert.NotEmpty(t, kms.Services)
 				assert.NotEmpty(t, kms.Services)
 				assert.NotEmpty(t, kms.PersistentVolumes)
 				assert.NotEmpty(t, kms.PersistentVolumes)
 				assert.NotEmpty(t, kms.PersistentVolumeClaims)
 				assert.NotEmpty(t, kms.PersistentVolumeClaims)
-				assert.NotEmpty(t, kms.DCGMDevices)
+				//assert.NotEmpty(t, kms.DCGMDevices)
 			},
 			},
 		},
 		},
 	}
 	}