From 533af5c322cfb305907a8fcc33355ddb9c40ebae Mon Sep 17 00:00:00 2001
From: wwhenxuan <wwhenxuan@gmail.com>
Date: Thu, 5 Mar 2026 17:37:29 +0800
Subject: [PATCH 1/2] whenxuan: update the version of s2generator for 0.0.9

---
 s2generator/__init__.py              | 2 +-
 s2generator/augmentation/__init__.py | 6 +++---
 setup.py                             | 2 +-
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/s2generator/__init__.py b/s2generator/__init__.py
index 1323b70..26f5949 100644
--- a/s2generator/__init__.py
+++ b/s2generator/__init__.py
@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 
-__version__ = "0.0.8"
+__version__ = "0.0.9"
 
 __all__ = [
     "Node",
diff --git a/s2generator/augmentation/__init__.py b/s2generator/augmentation/__init__.py
index 8538e6c..47df9db 100644
--- a/s2generator/augmentation/__init__.py
+++ b/s2generator/augmentation/__init__.py
@@ -9,7 +9,7 @@
 __all__ = [
     "amplitude_modulation",
     "censor_augmentation",
-    "empirical_model_modulation",
+    "empirical_mode_modulation",
     "frequency_perturbation",
     "spike_injection",
     "wiener_filter",
@@ -23,8 +23,8 @@
 # Import the censoring augmentation function
 from ._censor_augmentation import censor_augmentation
 
-# Import the empirical model modulation function
-from ._empirical_model_modulation import empirical_model_modulation
+# Import the empirical mode modulation function
+from ._empirical_mode_modulation import empirical_mode_modulation
 
 # Import the frequency perturbation function
 from ._frequency_perturbation import frequency_perturbation
diff --git a/setup.py b/setup.py
index 4de574f..075db71 100644
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@
 setuptools.setup(
     name="S2Generator",
     packages=setuptools.find_packages(),
-    version="0.0.8",
+    version="0.0.9",
     description="A series-symbol (S2) dual-modality data generation mechanism, enabling the unrestricted creation of high-quality time series data paired with corresponding symbolic representations.",  # 包的简短描述
     url="https://github.com/wwhenxuan/S2Generator",
     author="whenxuan, johnfan12, changewam",

From b84b2fe65054b7e825875a6e98287fb3799d6950 Mon Sep 17 00:00:00 2001
From: wwhenxuan <wwhenxuan@gmail.com>
Date: Thu, 5 Mar 2026 18:04:38 +0800
Subject: [PATCH 2/2] whenxuan: update the unit test

---
 s2generator/base.py        |  6 +++---
 tests/test_augmentation.py | 10 +++++++++-
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/s2generator/base.py b/s2generator/base.py
index 8c78568..69586b8 100644
--- a/s2generator/base.py
+++ b/s2generator/base.py
@@ -376,9 +376,9 @@ def val_diff(self, xs: ndarray, deterministic: Optional[bool] = True) -> ndarray
         if xs.ndim > 1:
             # For multivariate case, keep other dimensions constant
             x_uniform_input = np.tile(np.mean(xs, axis=0), (n_integration_points, 1))
-            x_uniform_input[
-                :, 0
-            ] = x_uniform  # Replace first dimension with uniform grid
+            x_uniform_input[:, 0] = (
+                x_uniform  # Replace first dimension with uniform grid
+            )
         else:
             x_uniform_input = x_uniform.reshape(-1, 1)  # Ensure 2D array for val method
 
diff --git a/tests/test_augmentation.py b/tests/test_augmentation.py
index 158f19d..a73af76 100644
--- a/tests/test_augmentation.py
+++ b/tests/test_augmentation.py
@@ -12,7 +12,11 @@
 from s2generator.augmentation import (
     amplitude_modulation,
     censor_augmentation,
+    empirical_mode_modulation,
     frequency_perturbation,
+    wiener_filter,
+    add_linear_trend,
+    time_series_mixup,
 )
 
 from s2generator.augmentation._frequency_perturbation import sample_random_perturbation
@@ -60,7 +64,11 @@ def test_frequency_perturbation(self) -> None:
         r = 0.3
 
         perturbed_series = frequency_perturbation(
-            series=series, min_alpha=min_alpha, max_alpha=max_alpha, r=r, rng=self.rng
+            time_series=series,
+            min_alpha=min_alpha,
+            max_alpha=max_alpha,
+            r=r,
+            rng=self.rng,
         )
 
         # Check that the output has the same length as the input