Addressing comments: Enforce and clean up type hints

Author: sugatmahanti

src/llmcompressor/metrics/logger.py

@@ -223,7 +223,7 @@ def lambda_func(
 
     def log_hyperparams(
         self,
-        params: dict,
+        params: dict[str, float],
         level: int | str | None = None,
     ) -> bool:
         """
@@ -931,7 +931,7 @@ def log_scalar(
         value: float,
         step: int | None = None,
         wall_time: float | None = None,
-        log_types: str | list[str] | None = ALL_TOKEN,
+        log_types: str | list[str] = ALL_TOKEN,
         level: int | str | None = None,
     ):
         """
@@ -962,7 +962,7 @@ def log_scalars(
         values: dict[str, float],
         step: int | None = None,
         wall_time: float | None = None,
-        log_types: str | list[str] | None = ALL_TOKEN,
+        log_types: str | list[str] = ALL_TOKEN,
         level: int | str | None = None,
     ):
         """
@@ -990,7 +990,7 @@ def log_scalars(
     def log_hyperparams(
         self,
         params: dict,
-        log_types: str | list[str] | None = ALL_TOKEN,
+        log_types: str | list[str] = ALL_TOKEN,
         level: int | str | None = None,
     ):
         """
@@ -1210,7 +1210,7 @@ class MetricLoggingWrapper(LoggingWrapperBase):
 
     def log_hyperparams(
         self,
-        params: dict,
+        params: dict[str, float],
         log_types: str | list[str] = ALL_TOKEN,
         level: int | str | None = None,
     ):

src/llmcompressor/metrics/utils/frequency_manager.py

@@ -256,7 +256,7 @@ def log_ready(
     current_log_step: LogStepType | None,
     last_log_step: LogStepType | None,
     log_frequency: LogStepType | None,
-    last_model_update_step: LogStepType | None = None,
+    last_model_update_step: LogStepType = None,
     check_model_update: bool = False,
 ):
     """