From ed1173a88c68abef0db7110bde3e607d629eaeac Mon Sep 17 00:00:00 2001
From: Sebastian Schmidt <mrschmidt@google.com>
Date: Tue, 2 Jun 2020 13:44:00 -0700
Subject: [PATCH] Transaction/WriteBatch signature fix

---
 packages/firestore/exp/index.d.ts | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/packages/firestore/exp/index.d.ts b/packages/firestore/exp/index.d.ts
index 27b0891f675..03747f48771 100644
--- a/packages/firestore/exp/index.d.ts
+++ b/packages/firestore/exp/index.d.ts
@@ -172,7 +172,11 @@ export class Transaction {
   set<T>(
     documentRef: DocumentReference<T>,
     data: T,
-    options?: SetOptions
+  ): Transaction;
+  set<T>(
+    documentRef: DocumentReference<T>,
+    data: Partial<T>,
+    options: SetOptions
   ): Transaction;
 
   update(documentRef: DocumentReference<any>, data: UpdateData): Transaction;
@@ -192,7 +196,11 @@ export class WriteBatch {
   set<T>(
     documentRef: DocumentReference<T>,
     data: T,
-    options?: SetOptions
+  ): WriteBatch;
+  set<T>(
+    documentRef: DocumentReference<T>,
+    data: Partial<T>,
+    options: SetOptions
   ): WriteBatch;
 
   update(documentRef: DocumentReference<any>, data: UpdateData): WriteBatch;