From c5eed797ec585262a00339e6537b5ef629911ada Mon Sep 17 00:00:00 2001
From: UpstreamData <brett@upstreamdata.ca>
Date: Thu, 25 Jan 2024 10:07:19 -0700
Subject: [PATCH] refactor: update type annotations in config.

---
 pyasic/config/base.py        | 2 +-
 pyasic/config/fans.py        | 2 +-
 pyasic/config/mining.py      | 2 +-
 pyasic/config/pools.py       | 2 +-
 pyasic/config/temperature.py | 4 ++--
 5 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/pyasic/config/base.py b/pyasic/config/base.py
index 0d416acd..4a97e988 100644
--- a/pyasic/config/base.py
+++ b/pyasic/config/base.py
@@ -70,7 +70,7 @@ class MinerConfigValue:
     def from_dict(cls, dict_conf: Union[dict, None]):
         return cls()
 
-    def as_dict(self):
+    def as_dict(self) -> dict:
         return asdict(self)
 
     def as_am_modern(self) -> dict:
diff --git a/pyasic/config/fans.py b/pyasic/config/fans.py
index 82565596..d2202853 100644
--- a/pyasic/config/fans.py
+++ b/pyasic/config/fans.py
@@ -35,7 +35,7 @@ class FanModeNormal(MinerConfigValue):
         return cls(**cls_conf)
 
     @classmethod
-    def from_vnish(cls, web_cooling_settings: dict):
+    def from_vnish(cls, web_cooling_settings: dict) -> "FanModeNormal":
         cls_conf = {}
         if web_cooling_settings.get("fan_min_count") is not None:
             cls_conf["minimum_fans"] = web_cooling_settings["fan_min_count"]
diff --git a/pyasic/config/mining.py b/pyasic/config/mining.py
index 4e29c049..23231428 100644
--- a/pyasic/config/mining.py
+++ b/pyasic/config/mining.py
@@ -91,7 +91,7 @@ class MiningModeHPM(MinerConfigValue):
     def from_dict(cls, dict_conf: Union[dict, None]) -> "MiningModeHPM":
         return cls()
 
-    def as_am_modern(self):
+    def as_am_modern(self) -> dict:
         return {"miner-mode": "0"}
 
     def as_wm(self) -> dict:
diff --git a/pyasic/config/pools.py b/pyasic/config/pools.py
index 4d8eb443..4e8a738a 100644
--- a/pyasic/config/pools.py
+++ b/pyasic/config/pools.py
@@ -219,7 +219,7 @@ class PoolGroup(MinerConfigValue):
     def as_goldshell(self, user_suffix: str = None) -> list:
         return [pool.as_goldshell(user_suffix) for pool in self.pools]
 
-    def as_avalon(self, user_suffix: str = None) -> dict:
+    def as_avalon(self, user_suffix: str = None) -> str:
         if len(self.pools) > 0:
             return self.pools[0].as_avalon(user_suffix=user_suffix)
         return Pool("", "", "").as_avalon()
diff --git a/pyasic/config/temperature.py b/pyasic/config/temperature.py
index cdde2829..76b11970 100644
--- a/pyasic/config/temperature.py
+++ b/pyasic/config/temperature.py
@@ -72,7 +72,7 @@ class TemperatureConfig(MinerConfigValue):
         return cls(target=target_temp, danger=dangerous_temp)
 
     @classmethod
-    def from_vnish(cls, web_settings: dict):
+    def from_vnish(cls, web_settings: dict) -> "TemperatureConfig":
         try:
             if web_settings["miner"]["cooling"]["mode"]["name"] == "auto":
                 return cls(target=web_settings["miner"]["cooling"]["mode"]["param"])
@@ -81,7 +81,7 @@ class TemperatureConfig(MinerConfigValue):
         return cls()
 
     @classmethod
-    def from_boser(cls, grpc_miner_conf: dict):
+    def from_boser(cls, grpc_miner_conf: dict) -> "TemperatureConfig":
         try:
             temperature_conf = grpc_miner_conf["temperature"]
         except KeyError: