fix(test): Set correct type annotation for period_start and period_end

frappe · May 29, 2023 · 6dd6b2c · 6dd6b2c
1 parent 9a2e446
commit 6dd6b2c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/press/press/doctype/payout_order/payout_order.py b/press/press/doctype/payout_order/payout_order.py
@@ -10,6 +10,8 @@
 from press.press.doctype.invoice_item.invoice_item import InvoiceItem
 from press.press.doctype.payout_order_item.payout_order_item import PayoutOrderItem
 
+from datetime import date
+
 
 class PayoutOrder(Document):
 	def validate(self):
@@ -203,8 +205,8 @@ def get_unaccounted_marketplace_invoice_items():
 def create_payout_order_from_invoice_items(
 	invoice_items: List[InvoiceItem],
 	recipient: str,
-	period_start: str,
-	period_end: str,
+	period_start: date,
+	period_end: date,
 	mode_of_payment: str = "Cash",
 	notes: str = "",
 	type: str = "Marketplace",

diff --git a/press/press/doctype/subscription/test_subscription.py b/press/press/doctype/subscription/test_subscription.py
@@ -28,7 +28,6 @@ def create_test_subscription(
 
 
 class TestSubscription(unittest.TestCase):
-
 	def setUp(self):
 		self.team = create_test_team()