lecrapaud 0.14.3__tar.gz → 0.14.4__tar.gz

{lecrapaud-0.14.3 → lecrapaud-0.14.4}/PKG-INFO

@@ -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 → lecrapaud-0.14.4}/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 → lecrapaud-0.14.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lecrapaud"
-version = "0.14.3"
+version = "0.14.4"
 description = "Framework for machine and deep learning, with regression, classification and time series analysis"
 authors = [
     {name = "Pierre H. Gallet"}