pythagoras 0.20.3__py3-none-any.whl → 0.20.4__py3-none-any.whl

pythagoras/_070_protected_code_portals/package_manager.py

@@ -3,12 +3,43 @@ import importlib
 import sys
 from typing import Optional
 
+_uv_and_pip_installation_needed:bool = True
+
+def _install_uv_and_pip() -> None:
+    global _uv_and_pip_installation_needed
+    if not _uv_and_pip_installation_needed:
+        return
+    _uv_pip_installation_attempted = False
+
+    try:
+        importlib.import_module("uv")
+    except:
+        install_package("uv", use_uv=False)
+
+    try:
+        importlib.import_module("pip")
+    except:
+        install_package("pip", use_uv=True)
+
+
 def install_package(package_name:str
         , upgrade:bool=False
         , version:Optional[str]=None
+        , use_uv:bool = True
         ) -> None:
     """Install package using pip."""
-    command = [sys.executable, "-m", "uv", "pip", "install"]
+
+    if package_name == "pip":
+        assert use_uv
+    elif package_name == "uv":
+        assert not use_uv
+    else:
+        _install_uv_and_pip()
+
+    if use_uv:
+        command = [sys.executable, "-m", "uv", "pip", "install"]
+    else:
+        command = [sys.executable, "-m", "pip", "install"]
 
     if upgrade:
         command += ["--upgrade"]
@@ -21,10 +52,17 @@ def install_package(package_name:str
 
     importlib.import_module(package_name)
 
-def uninstall_package(package_name:str)->None:
-    """Uninstall package using pip."""
 
-    command = [sys.executable, "-m", "uv", "pip", "uninstall", package_name]
+def uninstall_package(package_name:str, use_uv:bool=True)->None:
+    """Uninstall package using uv or pip."""
+
+    assert package_name not in ["pip", "uv"]
+
+    if use_uv:
+        command = [sys.executable, "-m", "uv", "pip", "uninstall", package_name]
+    else:
+        command = [sys.executable, "-m", "pip", "uninstall", "-y", package_name]
+
     subprocess.run(command, check=True, stdout=subprocess.PIPE
         , stderr=subprocess.STDOUT, text=True)
 

{pythagoras-0.20.3.dist-info → pythagoras-0.20.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pythagoras
-Version: 0.20.3
+Version: 0.20.4
 Summary: Planet-scale distributed computing in Python.
 Keywords: cloud,ML,AI,serverless,distributed,parallel,machine-learning,deep-learning,pythagoras
 Author: Volodymyr (Vlad) Pavlov

{pythagoras-0.20.3.dist-info → pythagoras-0.20.4.dist-info}/RECORD

@@ -36,7 +36,7 @@ pythagoras/_070_protected_code_portals/RAM_guards.py,sha256=cd439db72375ad0c2c62
 pythagoras/_070_protected_code_portals/__init__.py,sha256=1e1a4ccccede6cd416d172967fb766ba91bbafddff078187799f044ed71677b3,195
 pythagoras/_070_protected_code_portals/fn_arg_names_checker.py,sha256=2000ba3cab7d5828eba5b096358249c3ed460f7b82d41aaabcad945fd5e48a74,1215
 pythagoras/_070_protected_code_portals/list_flattener.py,sha256=9a54b512ad9dc201db348e7908f5ca5f36d171ae3da433511d38023a6ba8d4b5,331
-pythagoras/_070_protected_code_portals/package_manager.py,sha256=004cdd409843efd5a252ff829edbf8e21234ab468eaa104daa0f7d6ff8d49baa,1122
+pythagoras/_070_protected_code_portals/package_manager.py,sha256=30adf8d75f55b9b2c587dc95f4aa1c2154fa9354d668af6f0d586bb42f0d5b17,2008
 pythagoras/_070_protected_code_portals/protected_decorators.py,sha256=9cede3734b0235e15a9534fcf393454f9107073f90cf9bd5a532696d277e35da,1412
 pythagoras/_070_protected_code_portals/protected_portal_core_classes.py,sha256=58e66f14598a23d6d4bfa4dd3d510971ce272a11ea73a22d11b72487df112968,6616
 pythagoras/_070_protected_code_portals/python_packages_guards.py,sha256=a4ff97c250cbc94ad379edcab510fe10d4d590de9733bbff437bed71c025cecb,1025
@@ -60,6 +60,6 @@ pythagoras/_900_project_stats_collector/__init__.py,sha256=e3b0c44298fc1c149afbf
 pythagoras/_900_project_stats_collector/project_analyzer.py,sha256=d06e9d7b516cb7424ef777e70abe9d5220e09b0b19476326b8974b4dc3917f89,3506
 pythagoras/__init__.py,sha256=94303c01a7bde4078fdbd90c0b142807a023fa352c473c3a544a2180b7254ae9,1062
 pythagoras/core/__init__.py,sha256=26640880921763e8801b74da28fccf1cd47ac0ba0d40ee8fcaf7d748410406bf,189
-pythagoras-0.20.3.dist-info/WHEEL,sha256=7de84e261f5edc1201bd668234a371ec3eb94db037d1f75c0139430c1319ab31,79
-pythagoras-0.20.3.dist-info/METADATA,sha256=16ba3e6e4cb7a927c474c6189796e6f264dd180ed3413014a002a34272fedc33,4177
-pythagoras-0.20.3.dist-info/RECORD,,
+pythagoras-0.20.4.dist-info/WHEEL,sha256=7de84e261f5edc1201bd668234a371ec3eb94db037d1f75c0139430c1319ab31,79
+pythagoras-0.20.4.dist-info/METADATA,sha256=85100495f6fdc9536d83f64d0fee2ec62b5613c9b38edd80ef564e67f1ee348b,4177
+pythagoras-0.20.4.dist-info/RECORD,,