Skip to content

Commit fbce8c6

Browse files
committed
fix: gpu sync time in testing mode
1 parent 00a57cc commit fbce8c6

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

internal/controller/suite_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -195,7 +195,7 @@ var _ = BeforeSuite(func() {
195195
}).SetupWithManager(mgr)
196196
Expect(err).ToNot(HaveOccurred())
197197

198-
allocator = gpuallocator.NewGpuAllocator(ctx, mgr.GetClient(), 3*time.Second)
198+
allocator = gpuallocator.NewGpuAllocator(ctx, mgr.GetClient(), 150*time.Millisecond)
199199
_, err = allocator.SetupWithManager(ctx, mgr)
200200
Expect(err).ToNot(HaveOccurred())
201201

internal/gpuallocator/gpuallocator_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ var _ = Describe("GPU Allocator", func() {
3636
var allocator *GpuAllocator
3737

3838
BeforeEach(func() {
39-
allocator = NewGpuAllocator(ctx, k8sClient, 3*time.Second)
39+
allocator = NewGpuAllocator(ctx, k8sClient, 150*time.Millisecond)
4040
readyCh, err := allocator.SetupWithManager(ctx, mgr)
4141
Expect(err).NotTo(HaveOccurred())
4242

0 commit comments

Comments
 (0)