core-lens 0.1.dev83__tar.gz → 0.1.dev86__tar.gz

{core_lens-0.1.dev83 → core_lens-0.1.dev86}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: core-lens
-Version: 0.1.dev83
+Version: 0.1.dev86
 Summary: Query, analyse, and visualise CoreStack's microwatershed and Earth science data through a clean, composable Python API.
 Project-URL: Homepage, https://github.com/ApoorvaKashyap/core-lens
 Project-URL: Issues, https://github.com/ApoorvaKashyap/core-lens/issues

{core_lens-0.1.dev83 → core_lens-0.1.dev86}/src/core_lens/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '0.1.dev83'
-__version_tuple__ = version_tuple = (0, 1, 'dev83')
+__version__ = version = '0.1.dev86'
+__version_tuple__ = version_tuple = (0, 1, 'dev86')
 
 __commit_id__ = commit_id = None

{core_lens-0.1.dev83 → core_lens-0.1.dev86}/src/core_lens/base/namespaces/plot.py

@@ -7,6 +7,18 @@ from typing import TYPE_CHECKING, Any
 
 
 class SubplotOn(Enum):
+    """Temporal dimensions to split plot data across.
+
+    Used to specify how data should be partitioned into subplots or map layers
+    based on temporal columns added during materialisation.
+
+    Attributes:
+        YEAR: Split data by year.
+        MONTH: Split data by month.
+        SEASON: Split data by meteorological season.
+        SEASON_YEAR: Split data by season and year.
+    """
+
     YEAR = "year"
     MONTH = "month"
     SEASON = "season"

{core_lens-0.1.dev83 → core_lens-0.1.dev86}/src/core_lens/base/namespaces/stats.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import warnings
 from enum import Enum
 from itertools import combinations
 from typing import TYPE_CHECKING, Any, cast
@@ -14,12 +15,30 @@ if TYPE_CHECKING:
 
 
 class CorrelateMethod(Enum):
+    """Correlation methods.
+
+    Attributes:
+        PEARSON: Pearson correlation coefficient.
+        SPEARMAN: Spearman rank correlation.
+        KENDALL: Kendall Tau correlation.
+    """
+
     PEARSON = "pearson"
     SPEARMAN = "spearman"
     KENDALL = "kendall"
 
 
 class TestMethod(Enum):
+    """Statistical hypothesis testing methods.
+
+    Attributes:
+        T_TEST: Student's t-test.
+        MANN_WHITNEY: Mann-Whitney U test.
+        WILCOXON: Wilcoxon signed-rank test.
+        KS: Kolmogorov-Smirnov test.
+        CHI_SQUARE: Chi-square test.
+    """
+
     __test__ = False
     T_TEST = "t-test"
     MANN_WHITNEY = "mann-whitney"
@@ -29,12 +48,29 @@ class TestMethod(Enum):
 
 
 class ChangeMethod(Enum):
+    """Methods for calculating change over time.
+
+    Attributes:
+        ABSOLUTE: Absolute difference between periods.
+        PERCENTAGE: Percentage change between periods.
+        TREND: Linear trend over time.
+    """
+
     ABSOLUTE = "absolute"
     PERCENTAGE = "percentage"
     TREND = "trend"
 
 
 class AnomalyCrossMethod(Enum):
+    """Methods for detecting cross-sectional anomalies.
+
+    Attributes:
+        ZSCORE: Z-score method.
+        IQR: Interquartile range method.
+        PERCENTILE: Percentile-based method.
+        THRESHOLD: Fixed threshold method.
+    """
+
     ZSCORE = "zscore"
     IQR = "iqr"
     PERCENTILE = "percentile"
@@ -42,12 +78,29 @@ class AnomalyCrossMethod(Enum):
 
 
 class AnomalyTsMethod(Enum):
+    """Methods for detecting time-series anomalies.
+
+    Attributes:
+        STL: Seasonal-Trend decomposition using LOESS.
+        CUSUM: Cumulative sum control chart.
+        MAD: Median Absolute Deviation.
+    """
+
     STL = "stl"
     CUSUM = "cusum"
     MAD = "mad"
 
 
 class SimilarityMethod(Enum):
+    """Methods for calculating similarity or distance.
+
+    Attributes:
+        EUCLIDEAN: Euclidean distance.
+        COSINE: Cosine similarity.
+        MAHALANOBIS: Mahalanobis distance.
+        MANHATTAN: Manhattan distance.
+    """
+
     EUCLIDEAN = "euclidean"
     COSINE = "cosine"
     MAHALANOBIS = "mahalanobis"
@@ -871,9 +924,11 @@ class StatsNamespace:
         tidx = ids.index(target)
 
         # z-score normalise before computing distances.
-        means = np.nanmean(mat, axis=0)
-        stds = np.nanstd(mat, axis=0, ddof=1)
-        stds[stds == 0] = 1.0
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore", category=RuntimeWarning)
+            means = np.nanmean(mat, axis=0)
+            stds = np.nanstd(mat, axis=0, ddof=1)
+        stds[np.isnan(stds) | (stds == 0)] = 1.0
         norm = (mat - means) / stds
         tvec = norm[tidx]