Change noise field to default_factory (#26)

Co-authored-by: Thomas Pinder <[email protected]>
amazon-science · Oct 23, 2024 · 311dbd4 · 311dbd4
1 parent 8cfdcfa
commit 311dbd4
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -13,7 +13,7 @@ jobs:
       matrix:
         # Select the Python versions to test against
         os: ["ubuntu-latest", "macos-latest"]
-        python-version: ["3.10", "3.11"]
+        python-version: ["3.10", "3.11", "3.12"]
       fail-fast: true
     steps:
       - name: Check out the code

diff --git a/src/causal_validation/__about__.py b/src/causal_validation/__about__.py
@@ -1,3 +1,3 @@
-__version__ = "0.0.8"
+__version__ = "0.0.9"
 
 __all__ = ["__version__"]
diff --git a/src/causal_validation/transforms/noise.py b/src/causal_validation/transforms/noise.py
@@ -1,4 +1,4 @@
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from typing import Tuple
 
 from jaxtyping import Float
@@ -18,7 +18,9 @@ class Noise(AdditiveTransform):
     Normal with 0 loc and 0.1 scale.
     """
 
-    noise_dist: TimeVaryingParameter = TimeVaryingParameter(sampling_dist=norm(0, 0.1))
+    noise_dist: TimeVaryingParameter = field(
+        default_factory=lambda: TimeVaryingParameter(sampling_dist=norm(0, 0.1))
+    )
     _slots: Tuple[str] = ("noise_dist",)
 
     def get_values(self, data: Dataset) -> Float[np.ndarray, "N D"]: