truss 0.11.12rc509__tar.gz → 0.11.12rc511__tar.gz

{truss-0.11.12rc509 → truss-0.11.12rc511}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truss
-Version: 0.11.12rc509
+Version: 0.11.12rc511
 Summary: A seamless bridge from model development to model delivery
 Project-URL: Repository, https://github.com/basetenlabs/truss
 Project-URL: Homepage, https://truss.baseten.co

{truss-0.11.12rc509 → truss-0.11.12rc511}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "truss"
-version = "0.11.12rc509"
+version = "0.11.12rc511"
 description = "A seamless bridge from model development to model delivery"
 authors = [
     { name = "Pankaj Gupta", email = "no-reply@baseten.co" },

{truss-0.11.12rc509 → truss-0.11.12rc511}/truss/contexts/image_builder/serving_image_builder.py

@@ -761,7 +761,7 @@ class ServingImageBuilder(ImageBuilder):
                 else:
                     user_id = 60000
                 # dockerfile_contents += f"\nENV APP_HOME=$HOME"
-                dockerfile_contents += f"\nCOPY --chown={user_id}:{user_id} ./bptr-manifest $HOME/bptr-manifest"
+                dockerfile_contents += f"\nCOPY --chown={user_id}:{user_id} ./bptr-manifest /static-bptr/static-bptr-manifest.json"
             docker_file_path = build_dir / MODEL_DOCKERFILE_NAME
             docker_file_path.write_text(dockerfile_contents)
             return

{truss-0.11.12rc509 → truss-0.11.12rc511}/truss/templates/shared/lazy_data_resolver.py

@@ -53,7 +53,18 @@ class LazyDataResolverV2:
         self._lock = Lock()
         self._start_time = time.time()
         self.logger = logger or logging.getLogger(__name__)
-        self._is_collected_by_user = not self.bptr_exists()
+        
+        # Debug logging for bptr manifest detection
+        self.logger.info(f"LazyDataResolverV2: Checking for bptr manifest in paths: {LAZY_DATA_RESOLVER_PATH}")
+        bptr_found = self.bptr_exists()
+        self.logger.info(f"LazyDataResolverV2: bptr_exists() = {bptr_found}")
+        if bptr_found:
+            for path in LAZY_DATA_RESOLVER_PATH:
+                if path.exists():
+                    self.logger.info(f"LazyDataResolverV2: Found bptr manifest at {path}")
+                    break
+        
+        self._is_collected_by_user = not bptr_found
         thread = Thread(target=self._prefetch_in_thread, daemon=True)
         thread.start()
 
@@ -97,13 +108,24 @@ class LazyDataResolverV2:
     def _fetch(self) -> Union[str, Exception]:
         """cached and locked method to fetch the data."""
         if not self.bptr_exists():
+            self.logger.info("LazyDataResolverV2: No bptr manifest found, skipping download")
             return ""  # no data to resolve
-        import truss_transfer
+        
+        self.logger.info("LazyDataResolverV2: bptr manifest found, attempting to import truss_transfer")
+        try:
+            import truss_transfer
+            self.logger.info("LazyDataResolverV2: truss_transfer imported successfully")
+        except ImportError as e:
+            self.logger.error(f"LazyDataResolverV2: Failed to import truss_transfer: {e}")
+            return e
 
         try:
-            return truss_transfer.lazy_data_resolve(str(self._data_dir))
+            self.logger.info(f"LazyDataResolverV2: Calling truss_transfer.lazy_data_resolve with data_dir: {self._data_dir}")
+            result = truss_transfer.lazy_data_resolve(str(self._data_dir))
+            self.logger.info(f"LazyDataResolverV2: truss_transfer.lazy_data_resolve completed successfully")
+            return result
         except Exception as e:
-            self.logger.error(f"Error occurred while fetching data: {e}")
+            self.logger.error(f"LazyDataResolverV2: Error occurred while fetching data: {e}")
             return e
 
     def raise_if_not_collected(self):