From 27368a9bd28a1b0f23376f686002a1b4cc8f930b Mon Sep 17 00:00:00 2001
From: UpstreamData <brett@upstreamdata.ca>
Date: Mon, 11 Dec 2023 13:48:26 -0700
Subject: [PATCH] bug: fix some issues, and remove unused imports.

---
 pyasic/config/__init__.py      | 2 --
 pyasic/config/base.py          | 1 -
 pyasic/config/power_scaling.py | 4 +++-
 3 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/pyasic/config/__init__.py b/pyasic/config/__init__.py
index 9fb75a2a..116b57bf 100644
--- a/pyasic/config/__init__.py
+++ b/pyasic/config/__init__.py
@@ -15,8 +15,6 @@
 # ------------------------------------------------------------------------------
 from dataclasses import asdict, dataclass
 
-import toml
-
 from pyasic.config.fans import FanModeConfig
 from pyasic.config.mining import MiningModeConfig
 from pyasic.config.pools import PoolConfig
diff --git a/pyasic/config/base.py b/pyasic/config/base.py
index b082228f..3a5bfe8d 100644
--- a/pyasic/config/base.py
+++ b/pyasic/config/base.py
@@ -13,7 +13,6 @@
 #  See the License for the specific language governing permissions and         -
 #  limitations under the License.                                              -
 # ------------------------------------------------------------------------------
-from abc import ABC, abstractmethod
 from dataclasses import asdict, dataclass
 from enum import Enum
 from typing import Union
diff --git a/pyasic/config/power_scaling.py b/pyasic/config/power_scaling.py
index e776b49c..7df2fffe 100644
--- a/pyasic/config/power_scaling.py
+++ b/pyasic/config/power_scaling.py
@@ -94,7 +94,9 @@ class PowerScalingEnabled(MinerConfigValue):
     mode: str = field(init=False, default="enabled")
     power_step: int = None
     minimum_power: int = None
-    shutdown_enabled: PowerScalingShutdown = None
+    shutdown_enabled: Union[
+        PowerScalingShutdownEnabled, PowerScalingShutdownDisabled
+    ] = None
 
     @classmethod
     def from_bosminer(cls, power_scaling_conf: dict) -> "PowerScalingEnabled":