Merge pull request #15 from ajithabhks/variance_bug_fix

ajithabhks · web-flow · commit cef8aef2eb3a · 2024-11-09T19:06:46.000-06:00
Solve invalid value in variance calculation
diff --git a/sigmt/__version__.py b/sigmt/__version__.py
@@ -1,4 +1,4 @@
 """
 Returns version of the package
 """
-__version__ = "2.0.1"
+__version__ = "2.0.2"
diff --git a/sigmt/core/robust_estimation.py b/sigmt/core/robust_estimation.py
@@ -379,7 +379,7 @@ def get_variance(self) -> None:
         else:
             db = 100
 
-        self.z1_var = np.sqrt(np.real(da))
-        self.z2_var = np.sqrt(np.real(db))
+        self.z1_var = np.sqrt(abs(np.real(da)))
+        self.z2_var = np.sqrt(abs(np.real(db)))
         #
         self.predicted_coherency = np.sqrt(coh)

Original file line number	Diff line number	Diff line change
`@@ -379,7 +379,7 @@ def get_variance(self) -> None:`
`379`	`379`	`else:`
`380`	`380`	`db = 100`
`381`	`381`
`382`		`- self.z1_var = np.sqrt(np.real(da))`
`383`		`- self.z2_var = np.sqrt(np.real(db))`
	`382`	`+ self.z1_var = np.sqrt(abs(np.real(da)))`
	`383`	`+ self.z2_var = np.sqrt(abs(np.real(db)))`
`384`	`384`	`#`
`385`	`385`	`self.predicted_coherency = np.sqrt(coh)`