SleePyPhases 0.6.1__tar.gz → 0.6.2__tar.gz

{sleepyphases-0.6.1 → sleepyphases-0.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SleePyPhases
-Version: 0.6.1
+Version: 0.6.2
 Summary: A framwork for creating deep learning pipelines for sleep data
 Home-page: https://gitlab.com/sleep-is-all-you-need/sleepyphases
 Author: Franz Ehrlich

{sleepyphases-0.6.1 → sleepyphases-0.6.2}/SleePyPhases/phases/ExtractFeatures.py

@@ -166,7 +166,7 @@ class ExtractFeatures(Phase):
         featureConfigs = self.getConfig("featureExtraction.features", [])
         if featureNameDict != {}:
             featureNames = list(featureNameDict.keys())
-            featureConfigs = [f for f in featureConfigs if f["name"] in featureNames]
+            featureConfigs = [f for f in featureConfigs if f["name"] + "_checked" in featureNames]
 
         recordLoader = RecordLoader.get()
         recordSignal = None
@@ -181,9 +181,10 @@ class ExtractFeatures(Phase):
 
             
         for featureConfig in featureConfigs:
+            checkedCol = featureConfig["name"] + "_checked"
             if recordSignal is None:
                 self.features.loc[recordId, "recordId"] = recordId
-                self.features.loc[recordId, featureConfig["name"]] = True
+                self.features.loc[recordId, checkedCol] = True
                 self.registerData("features", self.features)
                 return {}
             
@@ -198,7 +199,7 @@ class ExtractFeatures(Phase):
 
                 self.features.loc[idx, "recordId"] = recordId
                 self.features.loc[idx, "channel"] = c
-                self.features.loc[idx, featureConfig["name"]] = True
+                self.features.loc[idx, checkedCol] = True
         
         self.registerData("features", self.features)
         idx = (recordId, channel) if channel is not None else recordId
@@ -211,7 +212,7 @@ class ExtractFeatures(Phase):
         recordIds = metadata["recordId"]
         forceCalculation = self.getConfig("featureExtraction.force", False)
 
-        reqFeatures = {f["name"]: True for f in self.getConfig("featureExtraction.features", [])}
+        reqFeatures = {f["name"] + "_checked": True for f in self.getConfig("featureExtraction.features", [])}
 
         recordId = self.getConfig("recordId", False)
         recordIds = [recordId] if recordId else recordIds.iloc

{sleepyphases-0.6.1 → sleepyphases-0.6.2}/SleePyPhases.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SleePyPhases
-Version: 0.6.1
+Version: 0.6.2
 Summary: A framwork for creating deep learning pipelines for sleep data
 Home-page: https://gitlab.com/sleep-is-all-you-need/sleepyphases
 Author: Franz Ehrlich

{sleepyphases-0.6.1 → sleepyphases-0.6.2}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setuptools.setup(
     name="SleePyPhases",
-    version="v0.6.1"[1:],
+    version="v0.6.2"[1:],
     author="Franz Ehrlich",
     author_email="fehrlichd@gmail.com",
     description="A framwork for creating deep learning pipelines for sleep data",