more linting

This commit is contained in:
Shannon Sands 2025-05-27 13:11:36 +10:00
parent 89b38a233b
commit 54e574d350

View file

@ -131,7 +131,6 @@ class McpEnv(BaseEnv):
# full_dataset = full_dataset.shuffle(seed=42) # shuffle here
full_dataset = full_dataset.shuffle(seed=42) # Shuffling datasets object
# Create train/test split on the fly (e.g., 95% train, 5% test)
split_dataset = full_dataset.train_test_split(test_size=0.02, seed=42)