ygg 0.1.41__tar.gz → 0.1.43__tar.gz

{ygg-0.1.41 → ygg-0.1.43}/PKG-INFO

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ygg
-Version: 0.1.41
+Version: 0.1.43
 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.41 → ygg-0.1.43}/src/yggdrasil/databricks/compute/cluster.py

@@ -390,6 +390,8 @@ class Cluster(WorkspaceService):
 
             sleep_time = min(max_sleep_time, sleep_time * backoff)
 
+        self.wait_installed_libraries()
+
         self.raise_for_status()
 
         return self
@@ -816,7 +818,6 @@ class Cluster(WorkspaceService):
             if wait_timeout:
                 self.clusters_client().start(cluster_id=self.cluster_id)
                 self.wait_for_status(timeout=wait_timeout.total_seconds())
-                self.wait_installed_libraries(timeout=wait_timeout)
             else:
                 self.clusters_client().start(cluster_id=self.cluster_id)
 
@@ -835,7 +836,7 @@ class Cluster(WorkspaceService):
         if self.is_running:
             logger.info("Restarting %s", self)
             self.details = self.clusters_client().restart_and_wait(cluster_id=self.cluster_id)
-            return self.wait_installed_libraries()
+            return self.wait_for_status()
 
         return self.start()
 

{ygg-0.1.41 → ygg-0.1.43}/src/yggdrasil/databricks/compute/execution_context.py

@@ -149,11 +149,20 @@ meta["version_info"] = current_env.version_info
 
 print(json.dumps(meta))"""
 
-                content = self.execute_command(
-                    command=cmd,
-                    result_tag="<<RESULT>>",
-                    print_stdout=False,
-                )
+                try:
+                    content = self.execute_command(
+                        command=cmd,
+                        result_tag="<<RESULT>>",
+                        print_stdout=False,
+                    )
+                except ImportError:
+                    self.cluster.ensure_running()
+
+                    content = self.execute_command(
+                        command=cmd,
+                        result_tag="<<RESULT>>",
+                        print_stdout=False,
+                    )
 
                 self._remote_metadata = RemoteMetadata(**json.loads(content))
 

ygg-0.1.43/src/yggdrasil/version.py

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

ygg-0.1.41/src/yggdrasil/version.py

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