just update the Target drift attribute to so we dont have to inject the drift_percent every single time

2022-03-17 22:48:22 +00:00
parent e4f51b8a3f
commit bd61fb7e81
3 changed files with 15 additions and 8 deletions
--- a/app/models/objective_function.py
+++ b/app/models/objective_function.py
@ -30,6 +30,12 @@ class ObjectiveFunction(BaseModel):
        print(self.tcc_targets)
        return amount

+    def update_targets_drift(self, amount: float = 0.0):
+        for target in self.tif_targets:
+            target.drift = round(amount, 2)
+        for target in self.tcc_targets:
+            target.drift = round(amount, 2)
+
    def minimum_drift(self) -> float:
        minimum_drift = 0.0