From c6e7b0826c8b6cfee2b261ad86cc0ea410ecfeb8 Mon Sep 17 00:00:00 2001
From: Igal Klebanov <igalklebanov@gmail.com>
Date: Wed, 16 Nov 2022 23:56:02 +0200
Subject: [PATCH] apply same restrictions for `required(...)` & `partial(...)`
 masks.

---
 deno/lib/types.ts | 8 ++++----
 src/types.ts      | 8 ++++----
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/deno/lib/types.ts b/deno/lib/types.ts
index 7e8e5be23..685f940f5 100644
--- a/deno/lib/types.ts
+++ b/deno/lib/types.ts
@@ -2059,8 +2059,8 @@ export class ZodObject<
     UnknownKeys,
     Catchall
   >;
-  partial<Mask extends { [k in keyof T]?: true }>(
-    mask: Mask
+  partial<Mask extends objectKeyMask<T>>(
+    mask: optionalPickWith<Mask, objectKeyMask<T>>
   ): ZodObject<
     objectUtil.noNever<{
       [k in keyof T]: k extends keyof Mask ? ZodOptional<T[k]> : T[k];
@@ -2100,8 +2100,8 @@ export class ZodObject<
     UnknownKeys,
     Catchall
   >;
-  required<Mask extends { [k in keyof T]?: true }>(
-    mask: Mask
+  required<Mask extends objectKeyMask<T>>(
+    mask: optionalPickWith<Mask, objectKeyMask<T>>
   ): ZodObject<
     objectUtil.noNever<{
       [k in keyof T]: k extends keyof Mask ? deoptional<T[k]> : T[k];
diff --git a/src/types.ts b/src/types.ts
index 622de4ac0..c7f11f588 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -2059,8 +2059,8 @@ export class ZodObject<
     UnknownKeys,
     Catchall
   >;
-  partial<Mask extends { [k in keyof T]?: true }>(
-    mask: Mask
+  partial<Mask extends objectKeyMask<T>>(
+    mask: optionalPickWith<Mask, objectKeyMask<T>>
   ): ZodObject<
     objectUtil.noNever<{
       [k in keyof T]: k extends keyof Mask ? ZodOptional<T[k]> : T[k];
@@ -2100,8 +2100,8 @@ export class ZodObject<
     UnknownKeys,
     Catchall
   >;
-  required<Mask extends { [k in keyof T]?: true }>(
-    mask: Mask
+  required<Mask extends objectKeyMask<T>>(
+    mask: optionalPickWith<Mask, objectKeyMask<T>>
   ): ZodObject<
     objectUtil.noNever<{
       [k in keyof T]: k extends keyof Mask ? deoptional<T[k]> : T[k];