fix test

determined-ai · Sep 23, 2024 · edc1848 · edc1848
1 parent 7763779
commit edc1848
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/master/internal/db/postgres_experiments_intg_test.go b/master/internal/db/postgres_experiments_intg_test.go
@@ -21,7 +21,6 @@ import (
 	"github.com/determined-ai/determined/master/pkg/model"
 	"github.com/determined-ai/determined/master/pkg/protoutils/protoconverter"
 	"github.com/determined-ai/determined/master/pkg/ptrs"
-	"github.com/determined-ai/determined/master/pkg/schemas"
 	"github.com/determined-ai/determined/master/pkg/schemas/expconf"
 	"github.com/determined-ai/determined/master/pkg/searcher"
 	"github.com/determined-ai/determined/proto/pkg/checkpointv1"
@@ -757,9 +756,14 @@ func TestDeleteExperiments(t *testing.T) {
 
 		// Create experiment snapshot
 		//nolint:exhaustruct
-		config := schemas.WithDefaults(expconf.SearcherConfig{
-			RawSingleConfig: &expconf.SingleConfig{},
-		})
+		config := expconf.SearcherConfig{
+			RawSingleConfig: &expconf.SingleConfigV0{
+				RawMaxLength: &expconf.LengthV0{
+					Unit:  expconf.Batches,
+					Units: 10,
+				},
+			},
+		}
 		searcher1 := searcher.NewSearcher(3, searcher.NewSearchMethod(config), nil)
 		_, err := searcher1.InitialOperations()
 		require.NoError(t, err)