From b9c5a5ff314a4f336067b035e6c57ed85c8be582 Mon Sep 17 00:00:00 2001
From: yasir <yasiramunandar@gmail.com>
Date: Thu, 8 Dec 2022 11:05:42 +0700
Subject: [PATCH] Fix

---
 database/filters_db.py |  1 +
 database/warn_db.py    | 11 ++++-------
 2 files changed, 5 insertions(+), 7 deletions(-)

diff --git a/database/filters_db.py b/database/filters_db.py
index f7275a0f..6688012b 100644
--- a/database/filters_db.py
+++ b/database/filters_db.py
@@ -1,4 +1,5 @@
 from database import dbname
+from typing import Dict, Union
 
 filtersdb = dbname.filters
 
diff --git a/database/warn_db.py b/database/warn_db.py
index 953f7779..06ce9086 100644
--- a/database/warn_db.py
+++ b/database/warn_db.py
@@ -31,10 +31,9 @@ async def add_warn(chat_id: int, name: str, warn: dict):
     warns = await get_warns(chat_id)
     warns[name] = warn
 
-    await warnsdb.update_one({"chat_id": chat_id}, {"$set": {
-        "warns": warns
-    }},
-                             upsert=True)
+    await warnsdb.update_one(
+        {"chat_id": chat_id}, {"$set": {"warns": warns}}, upsert=True
+    )
 
 
 async def remove_warns(chat_id: int, name: str) -> bool:
@@ -44,9 +43,7 @@ async def remove_warns(chat_id: int, name: str) -> bool:
         del warnsd[name]
         await warnsdb.update_one(
             {"chat_id": chat_id},
-            {"$set": {
-                "warns": warnsd
-            }},
+            {"$set": {"warns": warnsd}},
             upsert=True,
         )
         return True