fix: auto-load tzdata in pyodide (#2343)

Author: haakonvt

cognite/client/utils/_pyodide_helpers.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import logging
 import os
 import warnings
 from collections.abc import Callable, MutableMapping
@@ -16,6 +17,9 @@
     from cognite.client._http_client import HTTPClient, HTTPClientConfig
 
 
+logger = logging.getLogger(__name__)
+
+
 def patch_sdk_for_pyodide() -> None:
     # -------------------
     # Patch Pyodide related issues
@@ -56,6 +60,21 @@ def patch_sdk_for_pyodide() -> None:
     if os.getenv("COGNITE_FUSION_NOTEBOOK") is not None:
         global_config.default_client_config = FusionNotebookConfig()
 
+    # - We attempt to load the package 'tzdata' automatically, as pyodide can't read IANA timezone info from
+    #   the OS and thus need this extra package. We need the timezone info because we use zoneinfo.ZoneInfo
+    #   internally for e.g. datapoints and workflows.
+    #   Note: This convenience will only work in chromium-based browsers (as of Sept 2025)
+    try:
+        import micropip  # type: ignore [import-not-found]
+        from pyodide.ffi import run_sync  # type: ignore [import-not-found]
+
+        run_sync(micropip.install("tzdata"))
+    except Exception:
+        logger.debug(
+            "Could not load 'tzdata' package automatically in pyodide. You may need to do this manually:"
+            "import micropip; await micropip.install('tzdata')"
+        )
+
 
 def http_client__init__(
     self: HTTPClient,

tests/utils.py

@@ -41,6 +41,7 @@
 from cognite.client.data_classes.capabilities import Capability, LegacyCapability, UnknownAcl
 from cognite.client.data_classes.data_modeling import TypedEdge, TypedEdgeApply, TypedNode, TypedNodeApply
 from cognite.client.data_classes.data_modeling.data_types import ListablePropertyType
+from cognite.client.data_classes.data_modeling.ids import ContainerId, ViewId
 from cognite.client.data_classes.data_modeling.query import NodeResultSetExpression, Query
 from cognite.client.data_classes.datapoints import (
     _INT_AGGREGATES,
@@ -69,6 +70,7 @@
 if TYPE_CHECKING:
     import pandas
 
+
 REPO_ROOT = Path(__file__).resolve().parent.parent
 
 T_Type = TypeVar("T_Type", bound=type)
@@ -530,7 +532,11 @@ def create_value(self, type_: Any, var_name: str | None = None) -> Any:
         elif container_type is tuple:
             if any(arg is ... for arg in args):
                 return tuple(self.create_value(first_not_none) for _ in range(self._random.randint(1, 3)))
-            raise NotImplementedError(f"Tuple with multiple types is not supported. {self._error_msg}")
+            elif all(arg is str for arg in args):
+                return tuple(self._random_string(self._random.randint(0, 10)) for _ in range(len(args)))
+            raise NotImplementedError(
+                f"Tuple with multiple types is not supported. Add on the above line. {self._error_msg}"
+            )
 
         if var_name == "external_id" and type_ is str:
             return self._random_string(50, sample_from=string.ascii_uppercase + string.digits)
@@ -611,6 +617,8 @@ def _type_checking(cls) -> dict[str, type]:
         from cognite.client import CogniteClient
 
         return {
+            "ContainerId": ContainerId,
+            "ViewId": ViewId,
             "CogniteClient": CogniteClient,
             "NumpyDatetime64NSArray": npt.NDArray[np.datetime64],
             "NumpyUInt32Array": npt.NDArray[np.uint32],