Merge pull request #16 from ispras/issue15

#15: fix excessive lead selection

src/ecglib/data/datasets.py

@@ -84,13 +84,13 @@ def get_name(self, index: int) -> str:
         return str(Path(self.get_fpath(index)).stem)
 
     def read_ecg_record(
-        self, file_path, data_type, leads=[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]
-    ):
+        self, file_path, data_type
+        ):
         if data_type == "npz":
             ecg_record = np.load(file_path)["arr_0"].astype("float64")
             frequency = None
         elif data_type == "wfdb":
-            ecg_record, ann = wfdb.rdsamp(file_path, channels=leads)
+            ecg_record, ann = wfdb.rdsamp(file_path)
             ecg_record = ecg_record.T
             ecg_record = ecg_record.astype("float64")
             frequency = ann['fs']
@@ -142,7 +142,7 @@ def __getitem__(self, index):
         file_path = self.ecg_data.iloc[index]["fpath"]
 
         # data standartization (scaling, resampling, cuts off, normalization and padding/truncation)
-        ecg_record, _ = self.read_ecg_record(file_path, self.data_type, self.leads)
+        ecg_record, _ = self.read_ecg_record(file_path, self.data_type)
         full_ecg_record_info = EcgRecord(
             signal=ecg_record[self.leads, :],
             frequency=ecg_frequency,