ygg 0.1.37__tar.gz → 0.1.38__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ygg
-Version: 0.1.37
+Version: 0.1.38
 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.37 → ygg-0.1.38}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ygg"
-version = "0.1.37"
+version = "0.1.38"
 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.37 → ygg-0.1.38}/src/ygg.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ygg
-Version: 0.1.37
+Version: 0.1.38
 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.37 → ygg-0.1.38}/src/yggdrasil/databricks/sql/engine.py

@@ -309,7 +309,6 @@ class SQLEngine(WorkspaceService):
             engine=self,
             statement_id=response.statement_id,
             _response=response,
-            _response_refresh_time=time.time(),
             disposition=disposition,
         )
 

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

@@ -49,6 +49,10 @@ DONE_STATES = {
     StatementState.SUCCEEDED
 }
 
+FAILED_STATES = {
+    StatementState.FAILED, StatementState.CANCELED
+}
+
 __all__ = [
     "StatementResult"
 ]
@@ -123,16 +127,11 @@ class StatementResult:
         statement_execution = self.workspace.sdk().statement_execution
 
         if self._response is None:
-            self._response = (
-                statement_execution
-                .get_statement(self.statement_id)
-            )
-
-        if self._response.status.state not in DONE_STATES:
-            self._response = (
-                statement_execution
-                .get_statement(self.statement_id)
-            )
+            # Initialize
+            self._response = statement_execution.get_statement(self.statement_id)
+        elif self._response.status.state not in DONE_STATES:
+            # Refresh
+            self._response = statement_execution.get_statement(self.statement_id)
 
         return self._response
 
@@ -223,13 +222,7 @@ class StatementResult:
         Returns:
             True if the statement failed or was cancelled.
         """
-        if self.persisted:
-            return True
-
-        if self._response is None:
-            return False
-
-        return self._response.status.state in [StatementState.CANCELED, StatementState.FAILED]
+        return self.state in FAILED_STATES
 
     @property
     def persisted(self):

ygg-0.1.38/src/yggdrasil/version.py

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

ygg-0.1.37/src/yggdrasil/version.py

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