refactor

YieldsLabs · Jul 14, 2023 · d22c204 · d22c204
1 parent ebf52ec
commit d22c204
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/ta_lib/core/src/ops.rs b/ta_lib/core/src/ops.rs
@@ -79,7 +79,7 @@ impl Series<f64> {
     }
 
     pub fn neg(&self) -> Series<f64> {
-        self.fmap(|val| val.map(|v| if *v == 0.0 { *v } else { -*v }))
+        self.fmap(|val| val.map(|v| v.neg()))
     }
 }
 

diff --git a/ta_lib/core/src/series.rs b/ta_lib/core/src/series.rs
@@ -106,11 +106,9 @@ impl Series<f64> {
     }
 
     pub fn nz(&self, replacement: Option<f64>) -> Self {
-        let replacement = replacement.unwrap_or(0.0);
-
         self.fmap(|opt| match opt {
             Some(v) => Some(*v),
-            None => Some(replacement),
+            None => Some(replacement.unwrap_or(0.0)),
         })
     }