ygg 0.1.38__tar.gz → 0.1.39__tar.gz

{ygg-0.1.38 → ygg-0.1.39}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ygg
-Version: 0.1.38
+Version: 0.1.39
 Summary: Type-friendly utilities for moving data between Python objects, Arrow, Polars, Pandas, Spark, and Databricks
 Author: Yggdrasil contributors
 License:                                  Apache License

{ygg-0.1.38 → ygg-0.1.39}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ygg"
-version = "0.1.38"
+version = "0.1.39"
 description = "Type-friendly utilities for moving data between Python objects, Arrow, Polars, Pandas, Spark, and Databricks"
 readme = { file = "README.md", content-type = "text/markdown" }
 license = { file = "LICENSE" }

{ygg-0.1.38 → ygg-0.1.39}/src/ygg.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ygg
-Version: 0.1.38
+Version: 0.1.39
 Summary: Type-friendly utilities for moving data between Python objects, Arrow, Polars, Pandas, Spark, and Databricks
 Author: Yggdrasil contributors
 License:                                  Apache License

{ygg-0.1.38 → ygg-0.1.39}/src/yggdrasil/databricks/sql/statement_result.py

@@ -195,6 +195,7 @@ class StatementResult:
         Returns:
             The result manifest or None for Spark SQL results.
         """
+        self.wait()
         return self.response.manifest
 
     @property
@@ -204,6 +205,7 @@ class StatementResult:
         Returns:
             The statement result payload from the API.
         """
+        self.wait()
         return self.response.result
 
     @property
@@ -253,7 +255,6 @@ class StatementResult:
             self, self.disposition, Disposition.EXTERNAL_LINKS
         )
 
-        self.wait()
         result_data = self.result
         wsdk = self.workspace.sdk()
 
@@ -320,6 +321,8 @@ class StatementResult:
                 f"Statement {self.statement_id} {self.state}: " + " | ".join(parts)
             )
 
+        return self
+
     def wait(
         self,
         timeout: Optional[int] = None,
@@ -339,20 +342,21 @@ class StatementResult:
 
         start = time.time()
         poll_interval = poll_interval or 1
-        current = self
 
         while not self.done:
             # still running / queued / pending
             if timeout is not None and (time.time() - start) > timeout:
                 raise TimeoutError(
-                    f"Statement {current.statement_id} did not finish within {timeout} seconds "
-                    f"(last state={current})"
+                    f"Statement {self.statement_id} did not finish within {timeout} seconds "
+                    f"(last state={self.state})"
                 )
 
             poll_interval = max(10, poll_interval * 1.2)
             time.sleep(poll_interval)
 
-        return current
+        self.raise_for_status()
+
+        return self
 
     def arrow_schema(self):
         """Return the Arrow schema for the result.

ygg-0.1.39/src/yggdrasil/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.39"

ygg-0.1.38/src/yggdrasil/version.py

@@ -1 +0,0 @@
-__version__ = "0.1.38"