lecrapaud 0.14.3__py3-none-any.whl → 0.14.4__py3-none-any.whl

lecrapaud/db/models/experiment.py

@@ -100,19 +100,19 @@ class Experiment(Base):
     )
 
     def get_features(self, target_number: int):
-        feature_selections = self.feature_selections
-        target_id = [t for t in self.targets if t.name == f"TARGET_{target_number}"][
-            0
-        ].id
-        feature_selection = [
-            fs for fs in feature_selections if fs.target_id == target_id
-        ][0]
-        features = [f.name for f in feature_selection.features]
+        targets = [t for t in self.targets if t.name == f"TARGET_{target_number}"]
+        if targets:
+            target_id = targets[0].id
+            feature_selection = [
+                fs for fs in self.feature_selections if fs.target_id == target_id
+            ]
+            if feature_selection:
+                feature_selection = feature_selection[0]
+                features = [f.name for f in feature_selection.features]
+                return features
 
         # fallback to path if no features found
-        if len(features) == 0:
-            features = joblib.load(f"{self.path}/TARGET_{target_number}/features.pkl")
-
+        features = joblib.load(f"{self.path}/TARGET_{target_number}/features.pkl")
         return features
 
     def get_all_features(self, date_column: str = None, group_column: str = None):

{lecrapaud-0.14.3.dist-info → lecrapaud-0.14.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: lecrapaud
-Version: 0.14.3
+Version: 0.14.4
 Summary: Framework for machine and deep learning, with regression, classification and time series analysis
 License: Apache License
 Author: Pierre H. Gallet

{lecrapaud-0.14.3.dist-info → lecrapaud-0.14.4.dist-info}/RECORD

@@ -12,7 +12,7 @@ lecrapaud/db/alembic/versions/2025_06_25_1759-72aa496ca65b_.py,sha256=sBgPLvvqI_
 lecrapaud/db/alembic.ini,sha256=TXrZB4pWVLn2EUg867yp6paA_19vGeirO95mTPA3nbs,3699
 lecrapaud/db/models/__init__.py,sha256=Lhyw9fVLdom0Fc6yIP-ip8FjkU1EwVwjae5q2VM815Q,740
 lecrapaud/db/models/base.py,sha256=J9ew-0z_-tnWAwhVvOmVDys2R6jPF_oSca_ny6wpXQE,7606
-lecrapaud/db/models/experiment.py,sha256=29Ea0PfbiEvXx_0IhZBkd14PqY5v2Adx7LaW6c-6xfM,4135
+lecrapaud/db/models/experiment.py,sha256=x7-zTuCCecE401f3wQ9l6u8HyGOKORGc0b4_z-FMriU,4219
 lecrapaud/db/models/feature.py,sha256=5o77O2FyRObnLOCGNj8kaPSGM3pLv1Ov6mXXHYkmnYY,1136
 lecrapaud/db/models/feature_selection.py,sha256=epvOkU3eyhSXvEpRePd9Ylxl3feDvMUf_v6-jeU7fRE,3469
 lecrapaud/db/models/feature_selection_rank.py,sha256=Ydsb_rAT58FoSH13wkGjGPByzsjPx3DITXgJ2jgZmow,2198
@@ -38,7 +38,7 @@ lecrapaud/misc/test-gpu-transformers.ipynb,sha256=k6MBSs_Um1h4PykvE-LTBcdpbWLbIF
 lecrapaud/model_selection.py,sha256=5Q7KJ-oKDcW5ILlFzk4MjHyywB8HlNM6txCFNofku18,67796
 lecrapaud/search_space.py,sha256=-JkzuMhaomdwiWi4HvVQY5hiw3-oREemJA16tbwEIp4,34854
 lecrapaud/utils.py,sha256=JdBB1NvbNIx4y0Una-kSZdo1_ZEocc5hwyYFIZKHmGg,8305
-lecrapaud-0.14.3.dist-info/LICENSE,sha256=MImCryu0AnqhJE_uAZD-PIDKXDKb8sT7v0i1NOYeHTM,11350
-lecrapaud-0.14.3.dist-info/METADATA,sha256=WoPUnqxj35FUK2nddUMy7zAx2DhC_DwPdK8m677Ybtc,11047
-lecrapaud-0.14.3.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-lecrapaud-0.14.3.dist-info/RECORD,,
+lecrapaud-0.14.4.dist-info/LICENSE,sha256=MImCryu0AnqhJE_uAZD-PIDKXDKb8sT7v0i1NOYeHTM,11350
+lecrapaud-0.14.4.dist-info/METADATA,sha256=GZzcYaoTXl_lBYTUk97bJDdylGMPr7PahX-H-BqoKWg,11047
+lecrapaud-0.14.4.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+lecrapaud-0.14.4.dist-info/RECORD,,