Make conversation methods compatible with latest Pyrogram.

MrBotDeveloper · web-flow · commit a37f0d75d66a · 2022-04-26T10:49:28.000+05:30
diff --git a/pyrogram/methods/messages/wait_for_callback_query.py b/pyrogram/methods/messages/wait_for_callback_query.py
@@ -22,10 +22,9 @@
 
 from pyrogram import types
 from pyrogram.filters import Filter
-from pyrogram.scaffold import Scaffold
 
 
-class WaitForCallbackQuery(Scaffold):
+class WaitForCallbackQuery:
     async def wait_for_callback_query(
         self,
         chat_id: Union[int, str],
diff --git a/pyrogram/methods/messages/wait_for_message.py b/pyrogram/methods/messages/wait_for_message.py
@@ -22,10 +22,9 @@
 
 from pyrogram import types
 from pyrogram.filters import Filter
-from pyrogram.scaffold import Scaffold
 
 
-class WaitForMessage(Scaffold):
+class WaitForMessage:
     async def wait_for_message(
         self,
         chat_id: Union[int, str],
