zenml-nightly 0.58.2.dev20240625__py3-none-any.whl → 0.60.0.dev20240627__py3-none-any.whl

zenml/logging/step_logging.py

@@ -145,61 +145,64 @@ def fetch_logs(
         )
 
     artifact_store = _load_artifact_store(artifact_store_id, zen_store)
-    if not artifact_store.isdir(logs_uri):
-        return _read_file(logs_uri, offset, length)
-    else:
-        files = artifact_store.listdir(logs_uri)
-        if len(files) == 1:
-            return _read_file(
-                os.path.join(logs_uri, str(files[0])), offset, length
-            )
+    try:
+        if not artifact_store.isdir(logs_uri):
+            return _read_file(logs_uri, offset, length)
         else:
-            is_negative_offset = offset < 0
-            files.sort(reverse=is_negative_offset)
-
-            # search for the first file we need to read
-            latest_file_id = 0
-            for i, file in enumerate(files):
-                file_size: int = artifact_store.size(
-                    os.path.join(logs_uri, str(file))
-                )  # type: ignore[assignment]
-
-                if is_negative_offset:
-                    if file_size >= -offset:
-                        latest_file_id = -(i + 1)
-                        break
+            files = artifact_store.listdir(logs_uri)
+            if len(files) == 1:
+                return _read_file(
+                    os.path.join(logs_uri, str(files[0])), offset, length
+                )
+            else:
+                is_negative_offset = offset < 0
+                files.sort(reverse=is_negative_offset)
+
+                # search for the first file we need to read
+                latest_file_id = 0
+                for i, file in enumerate(files):
+                    file_size: int = artifact_store.size(
+                        os.path.join(logs_uri, str(file))
+                    )  # type: ignore[assignment]
+
+                    if is_negative_offset:
+                        if file_size >= -offset:
+                            latest_file_id = -(i + 1)
+                            break
+                        else:
+                            offset += file_size
                     else:
-                        offset += file_size
-                else:
-                    if file_size > offset:
-                        latest_file_id = i
+                        if file_size > offset:
+                            latest_file_id = i
+                            break
+                        else:
+                            offset -= file_size
+
+                # read the files according to pre-filtering
+                files.sort()
+                ret = []
+                for file in files[latest_file_id:]:
+                    ret.append(
+                        _read_file(
+                            os.path.join(logs_uri, str(file)),
+                            offset,
+                            length,
+                        )
+                    )
+                    offset = 0
+                    length -= len(ret[-1])
+                    if length <= 0:
+                        # stop further reading, if the whole length is already read
                         break
-                    else:
-                        offset -= file_size
-
-            # read the files according to pre-filtering
-            files.sort()
-            ret = []
-            for file in files[latest_file_id:]:
-                ret.append(
-                    _read_file(
-                        os.path.join(logs_uri, str(file)),
-                        offset,
-                        length,
+
+                if not ret:
+                    raise DoesNotExistException(
+                        f"Folder '{logs_uri}' is empty in artifact store "
+                        f"'{artifact_store.name}'."
                     )
-                )
-                offset = 0
-                length -= len(ret[-1])
-                if length <= 0:
-                    # stop further reading, if the whole length is already read
-                    break
-
-            if not ret:
-                raise DoesNotExistException(
-                    f"Folder '{logs_uri}' is empty in artifact store "
-                    f"'{artifact_store.name}'."
-                )
-            return "".join(ret)
+                return "".join(ret)
+    finally:
+        artifact_store.cleanup()
 
 
 class StepLogsStorage:

zenml/stack/stack_component.py

@@ -827,6 +827,10 @@ class StackComponent:
             f"Suspending provisioned resources not implemented for {self}."
         )
 
+    def cleanup(self) -> None:
+        """Cleans up the component after it has been used."""
+        pass
+
     def __repr__(self) -> str:
         """String representation of the stack component.
 

zenml/zen_server/dashboard/assets/{404-CDPQCl4D.js → 404-C1mcUujL.js}

@@ -1 +1 @@
-import{j as e}from"./@tanstack-CEbkxrhX.js";import{h as s,r as t}from"./index-QORVVTMN.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-CHBapDaj.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-BawkpTlr.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-DYIyhCfd.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};