update for mypy

zenml-io · Oct 19, 2023 · 61dc4f1 · 61dc4f1
1 parent 1883737
commit 61dc4f1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/mlstacks/analytics/client.py b/src/mlstacks/analytics/client.py
@@ -16,7 +16,7 @@
 import os
 from logging import getLogger
 from types import TracebackType
-from typing import Any, Dict, Optional, Type
+from typing import Any, Dict, Optional, Type, cast
 from uuid import uuid4
 
 import click
@@ -133,7 +133,7 @@ def get_analytics_user_id() -> Optional[str]:
         config_file = os.path.join(config_dir, CONFIG_FILENAME)
         if os.path.exists(config_file):
             yaml_dict = load_yaml_as_dict(config_file)
-            return yaml_dict.get("analytics_user_id", None)
+            return cast(str, yaml_dict.get("analytics_user_id", None))
         return None
 
     @staticmethod