primitive 0.1.55__tar.gz → 0.1.56__tar.gz

{primitive-0.1.55 → primitive-0.1.56}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: primitive
-Version: 0.1.55
+Version: 0.1.56
 Project-URL: Documentation, https://github.com//primitivecorp/primitive-cli#readme
 Project-URL: Issues, https://github.com//primitivecorp/primitive-cli/issues
 Project-URL: Source, https://github.com//primitivecorp/primitive-cli

{primitive-0.1.55 → primitive-0.1.56}/src/primitive/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2024-present Dylan Stein <dylan@primitive.tech>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "0.1.55"
+__version__ = "0.1.56"

{primitive-0.1.55 → primitive-0.1.56}/src/primitive/agent/uploader.py

@@ -1,5 +1,6 @@
 import typing
 import shutil
+import os
 from loguru import logger
 from pathlib import Path, PurePath
 from ..utils.cache import get_artifacts_cache
@@ -33,14 +34,26 @@ class Uploader:
         for job_cache in subdirs:
             job_run_id = job_cache.name
 
-            files = sorted(
-                [
-                    w_path / file
-                    for w_path, _, w_files in job_cache.walk()
-                    for file in w_files
-                ],
-                key=lambda p: p.stat().st_size,
-            )
+            files = None
+            has_walk = getattr(job_cache, "walk", None)
+            if has_walk:
+                files = sorted(
+                    [
+                        w_path / file
+                        for w_path, _, w_files in job_cache.walk()
+                        for file in w_files
+                    ],
+                    key=lambda p: p.stat().st_size,
+                )
+            else:
+                files = sorted(
+                    [
+                        Path(Path(w_path) / file)
+                        for w_path, _, w_files in os.walk(job_cache)
+                        for file in w_files
+                    ],
+                    key=lambda p: p.stat().st_size,
+                )
 
             file_ids = []
             for file in files: