zenml-nightly 0.58.0.dev20240529__py3-none-any.whl → 0.58.1.dev20240610__py3-none-any.whl

zenml/utils/source_utils.py

@@ -21,7 +21,7 @@ import site
 import sys
 from distutils.sysconfig import get_python_lib
 from pathlib import Path, PurePath
-from types import ModuleType
+from types import BuiltinFunctionType, FunctionType, ModuleType
 from typing import (
     Any,
     Callable,
@@ -46,6 +46,16 @@ NoneType = type(None)
 NoneTypeSource = Source(
     module=NoneType.__module__, attribute="NoneType", type=SourceType.BUILTIN
 )
+FunctionTypeSource = Source(
+    module=FunctionType.__module__,
+    attribute=FunctionType.__name__,
+    type=SourceType.BUILTIN,
+)
+BuiltinFunctionTypeSource = Source(
+    module=BuiltinFunctionType.__module__,
+    attribute=BuiltinFunctionType.__name__,
+    type=SourceType.BUILTIN,
+)
 
 _CUSTOM_SOURCE_ROOT: Optional[str] = None
 
@@ -62,10 +72,14 @@ def load(source: Union[Source, str]) -> Any:
     if isinstance(source, str):
         source = Source.from_import_path(source)
 
+    # The types of some objects don't exist in the `builtin` module
+    # so we need to manually handle it here
     if source.import_path == NoneTypeSource.import_path:
-        # The class of the `None` object doesn't exist in the `builtin` module
-        # so we need to manually handle it here
         return NoneType
+    elif source.import_path == FunctionTypeSource.import_path:
+        return FunctionType
+    elif source.import_path == BuiltinFunctionTypeSource.import_path:
+        return BuiltinFunctionType
 
     import_root = None
     if source.type == SourceType.CODE_REPOSITORY:
@@ -104,7 +118,14 @@ def load(source: Union[Source, str]) -> Any:
 
 
 def resolve(
-    obj: Union[Type[Any], Callable[..., Any], ModuleType, NoneType],
+    obj: Union[
+        Type[Any],
+        Callable[..., Any],
+        ModuleType,
+        FunctionType,
+        BuiltinFunctionType,
+        NoneType,
+    ],
     skip_validation: bool = False,
 ) -> Source:
     """Resolve an object.
@@ -120,10 +141,14 @@ def resolve(
     Returns:
         The source of the resolved object.
     """
-    if obj is NoneType:  # type: ignore[comparison-overlap]
-        # The class of the `None` object doesn't exist in the `builtin` module
-        # so we need to manually handle it here
+    # The types of some objects don't exist in the `builtin` module
+    # so we need to manually handle it here
+    if obj is NoneType:
         return NoneTypeSource
+    elif obj is FunctionType:
+        return FunctionTypeSource
+    elif obj is BuiltinFunctionType:
+        return BuiltinFunctionTypeSource
     elif isinstance(obj, ModuleType):
         module = obj
         attribute_name = None

zenml/zen_server/auth.py

@@ -238,8 +238,8 @@ def authenticate_credentials(
             decoded_token = JWTToken.decode_token(
                 token=access_token,
             )
-        except AuthorizationException:
-            error = "Authentication error: error decoding access token. You may need to rerun zenml connect."
+        except AuthorizationException as e:
+            error = f"Authentication error: error decoding access token: {e}."
             logger.exception(error)
             raise AuthorizationException(error)
 

zenml/zen_server/dashboard/assets/{404-DIgYOHj5.js → 404-D5p6PIdn.js}

@@ -1 +1 @@
-import{j as e}from"./@tanstack-GFKzItNf.js";import{h as s,r as t}from"./index-DKgtNG83.js";import{E as r}from"./EmptyState-CPJglRv0.js";import{H as a}from"./help-hEyYkC3q.js";import{L as o}from"./@react-router-CCpyfPW-.js";import"./@radix-BrEjOMvl.js";import"./@reactflow-CF_wPv-n.js";function d(){return e.jsx("div",{className:"flex min-h-screen w-full flex-col",children:e.jsx(r,{icon:e.jsx(a,{className:"h-[120px] w-[120px] fill-neutral-300"}),children:e.jsxs("div",{className:"text-center",children:[e.jsx("h1",{className:"mb-2 text-display-xs font-semibold",children:"We can't find the page you are looking for"}),e.jsx("p",{className:"text-lg text-theme-text-secondary",children:"You can try typing a different URL or we can bring you back to your Homepage."}),e.jsx("div",{className:"mt-5 flex justify-center",children:e.jsx(s,{size:"md",asChild:!0,children:e.jsx(o,{className:"w-min self-center whitespace-nowrap",to:t.home,children:e.jsx("span",{className:"px-0.5",children:"Go to Home"})})})})]})})})}export{d as default};
+import{j as e}from"./@tanstack-CEbkxrhX.js";import{h as s,r as t}from"./index-BhYPVFKa.js";import{E as r}from"./EmptyState-BkooiGtL.js";import{H as a}from"./help-B8rqCvqn.js";import{L as o}from"./@react-router-DYovave8.js";import"./@radix-C9DBgJhe.js";import"./@reactflow-CegZ5GV3.js";function d(){return e.jsx("div",{className:"flex min-h-screen w-full flex-col",children:e.jsx(r,{icon:e.jsx(a,{className:"h-[120px] w-[120px] fill-neutral-300"}),children:e.jsxs("div",{className:"text-center",children:[e.jsx("h1",{className:"mb-2 text-display-xs font-semibold",children:"We can't find the page you are looking for"}),e.jsx("p",{className:"text-lg text-theme-text-secondary",children:"You can try typing a different URL or we can bring you back to your Homepage."}),e.jsx("div",{className:"mt-5 flex justify-center",children:e.jsx(s,{size:"md",asChild:!0,children:e.jsx(o,{className:"w-min self-center whitespace-nowrap",to:t.home,children:e.jsx("span",{className:"px-0.5",children:"Go to Home"})})})})]})})})}export{d as default};