Skip to content

Commit 6a894fb

Browse files
committed
add type hinting to HBLOCK
1 parent 4b9efe1 commit 6a894fb

1 file changed

Lines changed: 12 additions & 11 deletions

File tree

spacv/spacv.py

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import numbers
22
import warnings
3+
from typing import Optional, Union
34

45
import geopandas as gpd
56
import numpy as np
@@ -69,17 +70,17 @@ class HBLOCK(BaseSpatialCV):
6970

7071
def __init__(
7172
self,
72-
tiles_x=5,
73-
tiles_y=5,
74-
shape="square",
75-
method="unique",
76-
buffer_radius=0,
77-
direction="diagonal",
78-
n_groups=5,
79-
data=None,
80-
n_sims=10,
81-
distance_metric="euclidean",
82-
random_state=None,
73+
tiles_x: int = 5,
74+
tiles_y: int = 5,
75+
shape: str = "square",
76+
method: str = "unique",
77+
buffer_radius: Union[int, float] = 0,
78+
direction: str = "diagonal",
79+
n_groups: int = 5,
80+
data: Optional[np.ndarray] = None,
81+
n_sims: int = 10,
82+
distance_metric: str = "euclidean",
83+
random_state: Optional[int] = None,
8384
):
8485
self.tiles_x = tiles_x
8586
self.tiles_y = tiles_y

0 commit comments

Comments
 (0)