minor

OmicsML · Feb 3, 2025 · 0af5f2e · 0af5f2e
1 parent 2cca105
commit 0af5f2e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/readme.md b/examples/readme.md
@@ -114,8 +114,8 @@ def evaluate_pipeline(tune_mode, pipeline_planer):
     # Initialize wandb
     wandb.init(settings=wandb.Settings(start_method='thread'))
 
-    # Load data
-    dataloader = ClusteringDataset(args.data_dir, args.dataset)
+    # Load data according to the task
+    dataloader = TaskDataset(args.data_dir, args.dataset)
     data = dataloader.load_data(cache=args.cache)
 
     # Apply preprocessing pipeline