truss 0.11.12rc509__py3-none-any.whl → 0.11.12rc510__py3-none-any.whl

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):

{truss-0.11.12rc509.dist-info → truss-0.11.12rc510.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truss
-Version: 0.11.12rc509
+Version: 0.11.12rc510
 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.dist-info → truss-0.11.12rc510.dist-info}/RECORD

@@ -107,7 +107,7 @@ truss/templates/server/common/tracing.py,sha256=XSTXNoRtV8vXwveJoX3H32go0JKnLmzn
 truss/templates/server/common/patches/whisper/patch.py,sha256=kDECQ-wmEpeAZFhUTQP457ofueeMsm7DgNy9tqinhJQ,2383
 truss/templates/shared/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truss/templates/shared/dynamic_config_resolver.py,sha256=75s42NFhQI5jL7BqlJH_UkuQS7ptbtFh13f2nh6X5Wo,920
-truss/templates/shared/lazy_data_resolver.py,sha256=pR0XVeb7GXWg1fhp5i-jVXja3z7ILOd49Fa43uLKbrU,6151
+truss/templates/shared/lazy_data_resolver.py,sha256=zRsGRF62NQUWtYkqkKx3A-F0912mN3ug6Dorkh2iJzs,7408
 truss/templates/shared/log_config.py,sha256=l9udyu4VKHZePlfK9LQEd5TOUUodPuehypsXRSUL4Ac,5411
 truss/templates/shared/secrets_resolver.py,sha256=3prDe3Q06NTmUEe7KCW-W4TD1CzGck9lpDG789209z4,2110
 truss/templates/shared/serialization.py,sha256=_WC_2PPkRi-MdTwxwjG8LKQptnHi4sANfpOlKWevqWc,3736
@@ -370,8 +370,8 @@ truss_train/deployment.py,sha256=lWWANSuzBWu2M4oK4qD7n-oVR1JKdmw2Pn5BJQHg-Ck,307
 truss_train/loader.py,sha256=0o66EjBaHc2YY4syxxHVR4ordJWs13lNXnKjKq2wq0U,1630
 truss_train/public_api.py,sha256=9N_NstiUlmBuLUwH_fNG_1x7OhGCytZLNvqKXBlStrM,1220
 truss_train/restore_from_checkpoint.py,sha256=8hdPm-WSgkt74HDPjvCjZMBpvA9MwtoYsxVjOoa7BaM,1176
-truss-0.11.12rc509.dist-info/METADATA,sha256=Uot1ZaH-UabixhB4iRkgBxp3-7tWZLw6SedkgbxMKE8,6683
-truss-0.11.12rc509.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-truss-0.11.12rc509.dist-info/entry_points.txt,sha256=-MwKfHHQHQ6j0HqIgvxrz3CehCmczDLTD-OsRHnjjuU,130
-truss-0.11.12rc509.dist-info/licenses/LICENSE,sha256=FTqGzu85i-uw1Gi8E_o0oD60bH9yQ_XIGtZbA1QUYiw,1064
-truss-0.11.12rc509.dist-info/RECORD,,
+truss-0.11.12rc510.dist-info/METADATA,sha256=Uwsbuea-zCQ1ovv465WHZP4s6ys9pLr18dWUwOdS9NI,6683
+truss-0.11.12rc510.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+truss-0.11.12rc510.dist-info/entry_points.txt,sha256=-MwKfHHQHQ6j0HqIgvxrz3CehCmczDLTD-OsRHnjjuU,130
+truss-0.11.12rc510.dist-info/licenses/LICENSE,sha256=FTqGzu85i-uw1Gi8E_o0oD60bH9yQ_XIGtZbA1QUYiw,1064
+truss-0.11.12rc510.dist-info/RECORD,,