geoai-py 0.3.0__tar.gz → 0.3.1__tar.gz

{geoai_py-0.3.0 → geoai_py-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: geoai-py
-Version: 0.3.0
+Version: 0.3.1
 Summary: A Python package for using Artificial Intelligence (AI) with geospatial data
 Author-email: Qiusheng Wu <giswqs@gmail.com>
 License: MIT License

{geoai_py-0.3.0 → geoai_py-0.3.1}/docs/installation.md

@@ -76,6 +76,14 @@ If you encounter issues with the sqlite package, you can update it using the fol
 mamba update -c conda-forge sqlite
 ```
 
+### Notes for Windows Users
+
+If you use mamba to install geoai, you may not have the latest version of torchgeo, which may cause issues when importing geoai. To fix this, you can install the latest version of torchgeo using the following command:
+
+```bash
+pip install -U torchgeo
+```
+
 ### 👩‍💻 Development Installation
 
 For contributing to GeoAI development, install directly from the source repository:

{geoai_py-0.3.0 → geoai_py-0.3.1}/geoai/__init__.py

@@ -2,7 +2,7 @@
 
 __author__ = """Qiusheng Wu"""
 __email__ = "giswqs@gmail.com"
-__version__ = "0.3.0"
+__version__ = "0.3.1"
 
 
 import os

{geoai_py-0.3.0 → geoai_py-0.3.1}/geoai/extract.py

@@ -7,7 +7,6 @@ import geopandas as gpd
 from tqdm import tqdm
 
 import cv2
-from torchgeo.datasets import NonGeoDataset
 from torchvision.models.detection import maskrcnn_resnet50_fpn
 import torchvision.transforms as T
 import rasterio
@@ -16,6 +15,13 @@ from rasterio.features import shapes
 from huggingface_hub import hf_hub_download
 from .preprocess import get_raster_stats
 
+try:
+    from torchgeo.datasets import NonGeoDataset
+except ImportError as e:
+    raise ImportError(
+        "Your torchgeo version is too old. Please upgrade to the latest version using 'pip install -U torchgeo'."
+    )
+
 
 class BuildingFootprintDataset(NonGeoDataset):
     """

{geoai_py-0.3.0 → geoai_py-0.3.1}/geoai/preprocess.py

@@ -18,10 +18,16 @@ import matplotlib.pyplot as plt
 from tqdm import tqdm
 from torchvision.transforms import RandomRotation
 from shapely.affinity import rotate
-import torchgeo
 import torch
 import cv2
 
+try:
+    import torchgeo
+except ImportError as e:
+    raise ImportError(
+        "Your torchgeo version is too old. Please upgrade to the latest version using 'pip install -U torchgeo'."
+    )
+
 
 def download_file(url, output_path=None, overwrite=False):
     """

{geoai_py-0.3.0 → geoai_py-0.3.1}/geoai/utils.py

@@ -12,9 +12,15 @@ import xarray as xr
 import rioxarray
 import rasterio as rio
 from torch.utils.data import DataLoader
-from torchgeo.datasets import RasterDataset, stack_samples, unbind_samples, utils
-from torchgeo.samplers import RandomGeoSampler, Units
-from torchgeo.transforms import indices
+
+try:
+    from torchgeo.datasets import RasterDataset, stack_samples, unbind_samples, utils
+    from torchgeo.samplers import RandomGeoSampler, Units
+    from torchgeo.transforms import indices
+except ImportError as e:
+    raise ImportError(
+        "Your torchgeo version is too old. Please upgrade to the latest version using 'pip install -U torchgeo'."
+    )
 
 
 def view_raster(
@@ -1039,3 +1045,38 @@ def adaptive_regularization(
         return gpd.GeoDataFrame(geometry=results, crs=building_polygons.crs)
     else:
         return results
+
+
+def install_package(package):
+    """Install a Python package.
+
+    Args:
+        package (str | list): The package name or a GitHub URL or a list of package names or GitHub URLs.
+    """
+    import subprocess
+
+    if isinstance(package, str):
+        packages = [package]
+    elif isinstance(package, list):
+        packages = package
+    else:
+        raise ValueError("The package argument must be a string or a list of strings.")
+
+    for package in packages:
+        if package.startswith("https"):
+            package = f"git+{package}"
+
+        # Execute pip install command and show output in real-time
+        command = f"pip install {package}"
+        process = subprocess.Popen(command.split(), stdout=subprocess.PIPE)
+
+        # Print output in real-time
+        while True:
+            output = process.stdout.readline()
+            if output == b"" and process.poll() is not None:
+                break
+            if output:
+                print(output.decode("utf-8").strip())
+
+        # Wait for process to complete
+        process.wait()

{geoai_py-0.3.0 → geoai_py-0.3.1}/geoai_py.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: geoai-py
-Version: 0.3.0
+Version: 0.3.1
 Summary: A Python package for using Artificial Intelligence (AI) with geospatial data
 Author-email: Qiusheng Wu <giswqs@gmail.com>
 License: MIT License

{geoai_py-0.3.0 → geoai_py-0.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "geoai-py"
-version = "0.3.0"
+version = "0.3.1"
 dynamic = [
     "dependencies",
 ]
@@ -47,7 +47,7 @@ universal = true
 
 
 [tool.bumpversion]
-current_version = "0.3.0"
+current_version = "0.3.1"
 commit = true
 tag = true