Fix typehints for update_chat_notifications

KurimuzonAkuma · KurimuzonAkuma · commit 24b46743a273 · 2023-11-22T17:58:29.000+03:00
diff --git a/pyrogram/methods/chats/update_chat_notifications.py b/pyrogram/methods/chats/update_chat_notifications.py
@@ -16,7 +16,7 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
-import datetime
+from datetime import datetime, timedelta
 from typing import Union
 
 import pyrogram
@@ -77,8 +77,8 @@ async def update_chat_notifications(
         if not mute_until:
             mute_until = utils.max_datetime() if mute else utils.zero_datetime()
 
-        if isinstance(mute_until, datetime.timedelta):
-            mute_until = datetime.datetime.now() + mute_until
+        if isinstance(mute_until, timedelta):
+            mute_until = datetime.now() + mute_until
 
         r = await self.invoke(
             raw.functions.account.UpdateNotifySettings(
