Skip to content

Commit caf475b

Browse files
committed
updated data loading
1 parent 1097c9e commit caf475b

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

ads/opctl/operator/lowcode/recommender/model/recommender_dataset.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66

77
import pandas as pd
88

9+
from ads.opctl.operator.lowcode.common.utils import load_data
910
from ..operator_config import RecommenderOperatorConfig
1011

1112

@@ -19,6 +20,6 @@ def __init__(self, config: RecommenderOperatorConfig):
1920
The recommender operator spec.
2021
"""
2122
spec = config.spec
22-
self.interactions: pd.DataFrame = pd.read_csv(spec.interactions_data.url)
23-
self.users: pd.DataFrame = pd.read_csv(spec.user_data.url)
24-
self.items: pd.DataFrame = pd.read_csv(spec.item_data.url)
23+
self.interactions: pd.DataFrame = load_data(getattr(spec, "interactions_data"))
24+
self.users: pd.DataFrame = load_data(getattr(spec, "user_data"))
25+
self.items: pd.DataFrame = load_data(getattr(spec, "item_data"))

0 commit comments

Comments
 (0)