rmm commited on
Commit
f824145
·
1 Parent(s): 0e9c1b3

feat: added member var for top predictions in the observation class

Browse files
Files changed (1) hide show
  1. src/input/input_observation.py +10 -0
src/input/input_observation.py CHANGED
@@ -56,6 +56,16 @@ class InputObservation:
56
  self.date_option = date_option
57
  self.time_option = time_option
58
  self.uploaded_filename = uploaded_filename
 
 
 
 
 
 
 
 
 
 
59
 
60
  def __str__(self):
61
  return f"Observation: {self.image}, {self.latitude}, {self.longitude}, {self.author_email}, {self.date}, {self.time}, {self.date_option}, {self.time_option}, {self.uploaded_filename}"
 
56
  self.date_option = date_option
57
  self.time_option = time_option
58
  self.uploaded_filename = uploaded_filename
59
+ self._top_predictions = []
60
+
61
+ def set_top_predictions(self, top_predictions:list):
62
+ self._top_predictions = top_predictions
63
+
64
+ # add a method to get the top predictions (property?)
65
+ @property
66
+ def top_predictions(self):
67
+ return self._top_predictions
68
+
69
 
70
  def __str__(self):
71
  return f"Observation: {self.image}, {self.latitude}, {self.longitude}, {self.author_email}, {self.date}, {self.time}, {self.date_option}, {self.time_option}, {self.uploaded_filename}"