colinahill / terrapyn Goto Github PK

def multiply_da(da, scalar, dim):
    """
    Multiply values in a DataArray along a specific dimension, where `scalar` must be the same length as the target dimension.
    """    
    multiplier = xr.DataArray(
        scalar,
        dims=dim,
        coords=[da[dim]]
    )
    return da * multiplier

Users/chill/software/terrapyn/terrapyn/indices/spi.py:75: RuntimeWarning: divide by zero encountered in double_scalars
  alpha = 1 / (4 * A) * (1 + np.sqrt(1 + 4 / 3 * A))
/Users/chill/software/terrapyn/terrapyn/indices/spi.py:72: RuntimeWarning: divide by zero encountered in log
  log_values = np.log(finite_values)
/Users/chill/software/terrapyn/terrapyn/indices/spi.py:75: RuntimeWarning: invalid value encountered in double_scalars
  alpha = 1 / (4 * A) * (1 + np.sqrt(1 + 4 / 3 * A))
/Users/chill/software/terrapyn/terrapyn/indices/spi.py:74: RuntimeWarning: divide by zero encountered in log
  A = np.log(mean_values) - np.mean(log_values)
/Users/chill/software/terrapyn/terrapyn/indices/spi.py:74: RuntimeWarning: invalid value encountered in double_scalars
  A = np.log(mean_values) - np.mean(log_values)
/Users/chill/software/anaconda3/envs/dev/lib/python3.9/site-packages/scipy/stats/_distn_infrastructure.py:1987: RuntimeWarning: divide by zero encountered in true_divide
  x = np.asarray((x - loc)/scale, dtype=dtyp)

def open_single_layer_tif_as_dataarray(filepath: str = None, variable_name: str = None, **kwargs) -> xr.DataArray:
    """
    Open a geotiff with 1 layer as an xr.DataArray

    Args:
        filepath: Path to tif file.
        variable_name: Variable name for data.
    
    Returns:
        DataArray with 'x' and 'y' renamed to 'lon' and 'lat'.
    """
    da = rxr.open_rasterio(filepath, chunks="auto", **kwargs)
    da = da.drop_vars(names="band").squeeze()
    da = da.rename({"x": "lon", "y": "lat"})
    da.name = variable_name
    return da

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
Cell In[6], line 1
----> 1 client = tp.dask_utils.create_cluster_and_client()

File ~/Library/Caches/pypoetry/virtualenvs/environmental-cluster-5we7EhFs-py3.10/lib/python3.10/site-packages/terrapyn/dask_utils.py:20, in create_cluster_and_client(n_workers, threads_per_worker, memory_limit, **kwargs)
      9 def create_cluster_and_client(n_workers: int = 4, threads_per_worker: int = 1, memory_limit: str = "1GB", **kwargs):
     10     """Create a local cluster and initiate a client
     11 
     12     Args:
   (...)
     18         Dask Client
     19     """
---> 20     client = dask.distributed.Client(
     21         n_workers=n_workers, threads_per_worker=threads_per_worker, memory_limit=memory_limit, **kwargs
     22     )
     23     print(f"Client dashboard: {client.dashboard_link}")
     24     return client

AttributeError: module 'dask' has no attribute 'distributed'