Revert "Do not allow unsound pick/omit"

This reverts commit 0f4ce73.

Author: colinhacks

packages/zod/src/v4/classic/tests/object.test.ts

@@ -607,25 +607,3 @@ test("safeExtend() on object with refinements should not throw", () => {
 
   expect(() => schema.safeExtend({ b: z.string() })).not.toThrow();
 });
-
-test("pick() on object with refinements should throw", () => {
-  const schema = z
-    .object({
-      a: z.string(),
-      b: z.number(),
-    })
-    .refine(() => true);
-
-  expect(() => schema.pick({ a: true })).toThrow("Invalid .pick() on object schemas containing refinements");
-});
-
-test("omit() on object with refinements should throw", () => {
-  const schema = z
-    .object({
-      a: z.string(),
-      b: z.number(),
-    })
-    .refine(() => true);
-
-  expect(() => schema.omit({ b: true })).toThrow("Invalid .omit() on object schemas containing refinements");
-});

packages/zod/src/v4/core/util.ts

@@ -592,10 +592,8 @@ export const BIGINT_FORMAT_RANGES: Record<checks.$ZodBigIntFormats, [bigint, big
 
 export function pick(schema: schemas.$ZodObject, mask: Record<string, unknown>): any {
   const currDef = schema._zod.def;
-  if ((currDef.checks ?? []).length > 0) {
-    throw new Error("Invalid .pick() on object schemas containing refinements");
-  }
-  const def = mergeDefs(currDef, {
+
+  const def = mergeDefs(schema._zod.def, {
     get shape() {
       const newShape: Writeable<schemas.$ZodShape> = {};
       for (const key in mask) {
@@ -617,13 +615,10 @@ export function pick(schema: schemas.$ZodObject, mask: Record<string, unknown>):
 
 export function omit(schema: schemas.$ZodObject, mask: object): any {
   const currDef = schema._zod.def;
-  if ((currDef.checks ?? []).length > 0) {
-    throw new Error("Invalid .omit() on object schemas containing refinements");
-  }
 
-  const def = mergeDefs(currDef, {
+  const def = mergeDefs(schema._zod.def, {
     get shape() {
-      const newShape: Writeable<schemas.$ZodShape> = { ...currDef.shape };
+      const newShape: Writeable<schemas.$ZodShape> = { ...schema._zod.def.shape };
       for (const key in mask) {
         if (!(key in currDef.shape)) {
           throw new Error(`Unrecognized key: "${key}"`);
@@ -649,7 +644,7 @@ export function extend(schema: schemas.$ZodObject, shape: schemas.$ZodShape): an
   const checks = schema._zod.def.checks;
   const hasChecks = checks && checks.length > 0;
   if (hasChecks) {
-    throw new Error("Invalid .extend() on object schemas containing refinements. Use .safeExtend() instead.");
+    throw new Error("Object schemas containing refinements cannot be extended. Use `.safeExtend()` instead.");
   }
 
   const def = mergeDefs(schema._zod.def, {