Skip to content

Commit

Permalink
update function name to avoid conflicts
Browse files Browse the repository at this point in the history
  • Loading branch information
ronylpatil committed Mar 18, 2024
1 parent bf4baae commit 22b6134
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/data/load_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
infologger.info('*** Executing: load_dataset.py ***')

# load data from given path and return df
def load_data(remote_loc: str) -> pd.DataFrame :
def extract_data(remote_loc: str) -> pd.DataFrame :
try :
# correct way to read data from drive
remote_loc = 'https://drive.google.com/uc?id=' + remote_loc.split('/')[-2]
Expand Down Expand Up @@ -40,7 +40,7 @@ def main() -> None :
# create dir if not present, else execute without any warning/error
output_path = home_dir.as_posix() + params['load_dataset']['raw_data']
pathlib.Path(output_path).mkdir(parents = True, exist_ok = True)
data = load_data(params['load_dataset']['drive_link'])
data = extract_data(params['load_dataset']['drive_link'])
save_data(data, output_path = output_path, file_name = params['load_dataset']['file_name'])
infologger.info('program terminated normally!')

Expand Down

0 comments on commit 22b6134

Please sign in to comment.