update name

kyryl-opens-ml · Jul 20, 2024 · 175de0e · 175de0e
1 parent 66a958f
commit 175de0e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/module-4/kubeflow_pipelines/kfp_inference_pipeline.py b/module-4/kubeflow_pipelines/kfp_inference_pipeline.py
@@ -17,9 +17,9 @@ def load_data(test_data: Output[Dataset]):
     import shutil
     from pathlib import Path
 
-    from classic_example.data import load_cola_data
+    from classic_example.data import load_sst2_data
 
-    load_cola_data(Path("/app/data"))
+    load_sst2_data(Path("/app/data"))
 
     shutil.move(Path("/app/data") / "test.csv", test_data.path)