ygg 0.1.46__tar.gz → 0.1.47__tar.gz

{ygg-0.1.46 → ygg-0.1.47}/PKG-INFO

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ygg
-Version: 0.1.46
+Version: 0.1.47
 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.46 → ygg-0.1.47}/src/yggdrasil/databricks/compute/execution_context.py

@@ -456,7 +456,37 @@ print(json.dumps(meta))"""
             timeout=timeout or dt.timedelta(minutes=20)
         )
 
-        return self._decode_result(result, result_tag=result_tag, print_stdout=print_stdout)
+        try:
+            return self._decode_result(result, result_tag=result_tag, print_stdout=print_stdout)
+        except ModuleNotFoundError as remote_module_error:
+            _MOD_NOT_FOUND_RE = re.compile(r"No module named ['\"]([^'\"]+)['\"]")
+            module_name = _MOD_NOT_FOUND_RE.search(str(remote_module_error))
+            module_name = module_name.group(1) if module_name else None
+            module_name = module_name.split(".")[0]
+
+            if module_name and "yggdrasil" not in module_name:
+                LOGGER.debug(
+                    "Installing missing module %s from local environment",
+                    module_name,
+                )
+
+                self.install_temporary_libraries(
+                    libraries=[module_name],
+                )
+
+                LOGGER.warning(
+                    "Installed missing module %s from local environment",
+                    module_name,
+                )
+
+                return self.execute_command(
+                    command=command,
+                    timeout=timeout,
+                    result_tag=result_tag,
+                    print_stdout=print_stdout
+                )
+
+            raise remote_module_error
 
     # ------------------------------------------------------------------
     # generic local → remote uploader, via remote python

ygg-0.1.47/src/yggdrasil/version.py

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

ygg-0.1.46/src/yggdrasil/version.py

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