Move this code to an independent function:
|
try: |
|
# Drop the labels from the dataframe if specified |
|
if drop_labels: |
|
# Data is now only extracted features |
|
raw_feature_data.drop(labels_to_drop, axis=1, inplace=True) |
|
except KeyError as e: |
|
logger.warning(f"{feature_file_path} does not have the labels {labels_to_drop} to drop.") |
|
# Skip to the next image type |
|
continue |
Move this code to an independent function:
readii/src/readii/io/loaders/features.py
Lines 68 to 76 in 36bf8e3