Fix infinite loop while fetching artifact store in logs storage class

src/zenml/client_lazy_loader.py

@@ -172,10 +172,10 @@ def evaluate_all_lazy_load_args_in_client_methods(
     """
     import inspect
 
-    def _evaluate_args(func: Callable[..., Any]) -> Any:
+    def _evaluate_args(
+        func: Callable[..., Any], is_instance_method: bool
+    ) -> Any:
         def _inner(*args: Any, **kwargs: Any) -> Any:
-            is_instance_method = "self" in inspect.getfullargspec(func).args
-
             args_ = list(args)
             if not is_instance_method:
                 from zenml.client import Client
@@ -201,7 +201,11 @@ def _inner(*args: Any, **kwargs: Any) -> Any:
 
     def _decorate() -> Type["Client"]:
         for name, fn in inspect.getmembers(cls, inspect.isfunction):
-            setattr(cls, name, _evaluate_args(fn))
+            setattr(
+                cls,
+                name,
+                _evaluate_args(fn, "self" in inspect.getfullargspec(fn).args),
+            )
         return cls
 
     return _decorate()

src/zenml/logging/step_logging.py

@@ -28,7 +28,6 @@
     _load_artifact_store,
     _load_file_from_artifact_store,
 )
-from zenml.client import Client
 from zenml.exceptions import DoesNotExistException
 from zenml.logger import get_logger
 from zenml.logging import (
@@ -212,6 +211,7 @@ class StepLogsStorage:
     def __init__(
         self,
         logs_uri: str,
+        artifact_store: "BaseArtifactStore",
         max_messages: int = STEP_LOGS_STORAGE_MAX_MESSAGES,
         time_interval: int = STEP_LOGS_STORAGE_INTERVAL_SECONDS,
         merge_files_interval: int = STEP_LOGS_STORAGE_MERGE_INTERVAL_SECONDS,
@@ -220,6 +220,7 @@ def __init__(
 
         Args:
             logs_uri: the URI of the log file or folder.
+            artifact_store: Artifact Store from the current step context
             max_messages: the maximum number of messages to save in the buffer.
             time_interval: the amount of seconds before the buffer gets saved
                 automatically.
@@ -237,22 +238,11 @@ def __init__(
         self.disabled_buffer: List[str] = []
         self.last_save_time = time.time()
         self.disabled = False
-        self._artifact_store: Optional["BaseArtifactStore"] = None
+        self.artifact_store = artifact_store
 
         # Immutable filesystems state
         self.last_merge_time = time.time()
 
-    @property
-    def artifact_store(self) -> "BaseArtifactStore":
-        """Returns the active artifact store.
-
-        Returns:
-            The active artifact store.
-        """
-        if self._artifact_store is None:
-            self._artifact_store = Client().active_stack.artifact_store
-        return self._artifact_store
-
     def write(self, text: str) -> None:
         """Main write method.
 
@@ -406,13 +396,18 @@ def merge_log_files(self, merge_all_files: bool = False) -> None:
 class StepLogsStorageContext:
     """Context manager which patches stdout and stderr during step execution."""
 
-    def __init__(self, logs_uri: str) -> None:
+    def __init__(
+        self, logs_uri: str, artifact_store: "BaseArtifactStore"
+    ) -> None:
         """Initializes and prepares a storage object.
 
         Args:
             logs_uri: the URI of the logs file.
+            artifact_store: Artifact Store from the current step context.
         """
-        self.storage = StepLogsStorage(logs_uri=logs_uri)
+        self.storage = StepLogsStorage(
+            logs_uri=logs_uri, artifact_store=artifact_store
+        )
 
     def __enter__(self) -> "StepLogsStorageContext":
         """Enter condition of the context manager.

src/zenml/orchestrators/step_launcher.py

@@ -167,7 +167,7 @@ def launch(self) -> None:
             )
 
             logs_context = step_logging.StepLogsStorageContext(
-                logs_uri=logs_uri
+                logs_uri=logs_uri, artifact_store=self._stack.artifact_store
             )  # type: ignore[assignment]
 
             logs_model = LogsRequest(

src/zenml/orchestrators/step_runner.py

@@ -135,7 +135,8 @@ def run(
         if step_logging_enabled and not redirected.get():
             if step_run.logs:
                 logs_context = StepLogsStorageContext(  # type: ignore[assignment]
-                    logs_uri=step_run.logs.uri
+                    logs_uri=step_run.logs.uri,
+                    artifact_store=self._stack.artifact_store,
                 )
             else:
                 logger.debug(