fix: default values for nvidia_smi query timeout (#822)

flashcatcloud · Mar 8, 2024 · ca97095 · ca97095
1 parent aa21b82
commit ca97095
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/inputs/nvidia_smi/nvidia_smi.go b/inputs/nvidia_smi/nvidia_smi.go
@@ -43,6 +43,9 @@ func (s *GPUStats) Init() error {
 	if s.NvidiaSmiCommand == "" {
 		return types.ErrInstancesEmpty
 	}
+	if s.QueryTimeOut == 0 {
+		s.QueryTimeOut = config.Duration(5 * time.Second)
+	}
 
 	qFieldsOrdered, qFieldToRFieldMap, err := s.buildQFieldToRFieldMap()
 	if err != nil {