diff --git a/model_creator.py b/model_creator.py index 555f6a4..398b3b9 100644 --- a/model_creator.py +++ b/model_creator.py @@ -14,23 +14,19 @@ def check_datasets_presence(): dataset_1 = "Spotify_Dataset.csv" dataset_2 = "spotify_songs.csv" - destination_folder = "datasets" + destination_folder = "artifacts" if not os.path.exists(destination_folder): - os.makedirs(destination_folder) - print(f"Utworzono folder: {destination_folder}") - else: - print(f"Folder {destination_folder} już istnieje.") - + raise FileNotFoundError(destination_folder + " folder not found") if dataset_1 in os.listdir("/."): shutil.move(dataset_1, destination_folder) elif dataset_1 not in os.listdir(destination_folder): - raise FileNotFoundError(dataset_1 + "not found") + raise FileNotFoundError(dataset_1 + " not found") if dataset_2 in os.listdir("/."): shutil.move(dataset_2, destination_folder) elif dataset_2 not in os.listdir(destination_folder): - raise FileNotFoundError(dataset_2 + "not found") + raise FileNotFoundError(dataset_2 + " not found") def datasets_preparation(): df_1 = pd.read_csv("datasets/spotify_songs.csv")