Skip to content

Commit

Permalink
fix previous commit
Browse files Browse the repository at this point in the history
Sueda Ciftci committed Jul 20, 2024
1 parent 1565aa9 commit aee7e2d
Showing 2 changed files with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
@@ -229,7 +229,7 @@ def train_mrdgp(
training_data = file_manager.load_training_data()

typer.echo("Loading testing data!")
test_data = file_manager.load_training_data()
test_data = file_manager.load_testing_data()

breakpoint()
training_dict, test_dict = generate_data(training_data, test_data)
Original file line number Diff line number Diff line change
@@ -69,7 +69,7 @@ def load_training_data(self) -> dict:
"""Load training data from the dataset directory."""
for dirpath, _, filenames in os.walk(self.input_dir):
# Check if 'training_dataset.pkl' exists in the current directory
if "test_dataset.pkl" in filenames:
if "training_dataset.pkl" in filenames:
# If found, load the data
file_path = os.path.join(dirpath, "training_dataset.pkl")
with open(file_path, "rb") as file:
@@ -84,7 +84,7 @@ def load_testing_data(self) -> dict:
# Check if 'training_dataset.pkl' exists in the current directory
if "test_dataset.pkl" in filenames:
# If found, load the data
file_path = os.path.join(dirpath, "testing_dataset.pkl")
file_path = os.path.join(dirpath, "test_dataset.pkl")
with open(file_path, "rb") as file:
return pickle.load(file)
raise FileNotFoundError(

0 comments on commit aee7e2d

Please sign in to comment.