Update iot_simulator.py

iot_simulator.py

@@ -1,26 +1,44 @@
 import random
-import time
-import numpy as np
-from typing import Dict, List
+from typing import Dict, Optional
 
 class IoTSimulator:
-    """Simulates a robotic arm with sensors."""
+    """
+    Simulates a robotic arm with four sensors:
+    - temperature
+    - vibration
+    - motor current
+    - position error
+    Supports fault injection for testing anomaly detection.
+    """
+
     def __init__(self, seed: int = 42):
+        """
+        Initialize the simulator with baseline values and fault mode.
+        Args:
+            seed: Random seed for reproducibility.
+        """
         random.seed(seed)
         self.step = 0
         self.base_temp = 35.0
         self.base_vibration = 0.1
         self.base_current = 2.0
         self.base_position_error = 0.01
-        self.fault_mode = None  # 'overheat', 'vibration', 'stall', 'drift'
+        self.fault_mode: Optional[str] = None  # 'overheat', 'vibration', 'stall', 'drift'
 
-    def set_fault(self, fault_type: str):
+    def set_fault(self, fault_type: Optional[str]):
+        """
+        Set the fault mode.
+        Args:
+            fault_type: One of 'overheat', 'vibration', 'stall', 'drift', or None.
+        """
         self.fault_mode = fault_type
 
-    def clear_fault(self):
-        self.fault_mode = None
-
     def read(self) -> Dict[str, float]:
+        """
+        Simulate a sensor reading, possibly with injected fault.
+        Returns:
+            Dictionary with sensor keys: temperature, vibration, motor_current, position_error.
+        """
         self.step += 1
         # Normal variation
         temp = self.base_temp + random.gauss(0, 0.5)