Update SafeDrones.py

Dependable-Intelligent-Systems-Lab · Sep 28, 2023 · 12a5b6c · 12a5b6c
1 parent 975ecc1
commit 12a5b6c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/SafeDrones/core/SafeDrones.py b/SafeDrones/core/SafeDrones.py
@@ -29,6 +29,9 @@ def __init__(self):
 
         # Mission Time
         self.time = None
+
+        self.danger_threshold = None
+        self.collision_threshold = None
 
         self.Set_Variables()
 
@@ -611,7 +614,7 @@ def GPS_Failure_Risk_Calc(self):
 
         return P_Fail, MTTF
 
-    def calculate_collision_risk(uav_1_trajectory, uav_2_trajectory, self.danger_threshold, self.collision_threshold):
+    def calculate_collision_risk(self, uav_1_trajectory, uav_2_trajectory):  # self.danger_threshold, self.collision_threshold
         # Check that both trajectories have the same length
         assert len(uav_1_trajectory) == len(uav_2_trajectory), "Trajectories must have the same length"