Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def generate_raw_db(self):
"""Generate the raw version of the MIT-BIH Atrial Fibrillation database in the 'raw' folder."""
print('Generating Raw MIT-BIH Atrial Fibrillation Database ...')
# Download database
wfdb.dl_database(self.db_name, self.raw_path)
# Get list of recordings
self.record_ids = [file.split('.')[0] for file in os.listdir(self.raw_path) if '.dat' in file]
print('Complete!\n')
def generate_raw_db(self):
"""Generate the raw version of the MIT-BIH Normal Sinus Rhythm database in the 'raw' folder."""
print('Generating Raw MIT-BIH Normal Sinus Rhythm Database ...')
# Download database
wfdb.dl_database(self.db_name, self.raw_path)
# Get list of recordings
self.record_ids = [file.split('.')[0] for file in os.listdir(self.raw_path) if '.dat' in file]
print('Complete!\n')