'Refactored by Sourcery'

retrofor · Jan 17, 2024 · f0a91b8 · f0a91b8
1 parent 26e8b45
commit f0a91b8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 14 deletions.
diff --git a/tests/plugins/HydroRoll/__init__.py b/tests/plugins/HydroRoll/__init__.py
@@ -20,9 +20,7 @@ def _init_directory(_prefix: str = ""):
 
 
 def _load_models():
-    models = {}
-    models["hola"] = joblib.load(join(HYDRO_DIR, "models", "hola.pkl"))
-    return models
+    return {"hola": joblib.load(join(HYDRO_DIR, "models", "hola.pkl"))}
 
 
 def load_model(model):

diff --git a/tests/plugins/HydroRoll/utils.py b/tests/plugins/HydroRoll/utils.py
@@ -59,9 +59,6 @@ async def rule(self) -> bool:
                     or self.event.group_id in self.config.accept_group
                 ):
                     return self.str_match(match_str)
-        elif self.config.handle_group_message:
-            if self.event.message_type == "guild":
-                return self.str_match(match_str)
         return False
 
     @abstractmethod
@@ -152,11 +149,9 @@ def roll_dice(
 
         if streamline:
             return str(total)
-        else:
-            if len(rolls) > int(threshold):
-                return str(total)
-            rolls_str = " + ".join(str(r) for r in rolls)
-            result_str = (
-                f"{total} = {rolls_str}" if is_reversed else f"{rolls_str} = {total}"
-            )
-            return result_str
+        if len(rolls) > int(threshold):
+            return str(total)
+        rolls_str = " + ".join(str(r) for r in rolls)
+        return (
+            f"{total} = {rolls_str}" if is_reversed else f"{rolls_str} = {total}"
+        )