Merge branch 'main' of https://github.com/aimclub/OCEANAI

aimclub · Dec 10, 2024 · fb1337e · fb1337e
2 parents 174a2ee + 240413d
commit fb1337e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/oceanai/modules/core/core.py b/oceanai/modules/core/core.py
@@ -384,6 +384,7 @@ def __post_init__(self):
         self._df_files_ranking: pd.DataFrame = pd.DataFrame()  # DataFrame с ранжированными данными
         # DataFrame с ранжированными предпочтениями на основе данных
         self._df_files_priority: pd.DataFrame = pd.DataFrame()
+        self._df_files_priority_score: List[List[float]] = []
         # DataFrame с ранжированными коллегами на основе данных
         self._df_files_colleague: pd.DataFrame = pd.DataFrame()
         self._df_files_priority_skill: pd.DataFrame = pd.DataFrame()
@@ -3856,6 +3857,8 @@ def _priority_calculation(
                 else:
                     try:
 
+                        self._df_files_priority_score = []
+
                         self._df_files_priority = self._df_files.copy()
                         df_files_priority = self._df_files.copy()
 
@@ -3893,6 +3896,8 @@ def _priority_calculation(
                                     for i in range(number_importance_traits)
                                 ],
                             ] = importance_traits
+
+                            self._df_files_priority_score.append(curr_weights[idx_max_values])
                     except Exception:
                         self._other_error(self._unknown_err, out=out)
                         return self._df_files_priority