Skip to content

Commit

Permalink
fix: use local dataset for CLV tests (#1412)
Browse files Browse the repository at this point in the history
- Modified test_utils.py to use local CLV dataset instead of downloading from GitHub.
- Ensured the dataset is included in the repository.
  • Loading branch information
nayan2167 authored Jan 19, 2025
1 parent 8a0b076 commit 6f12316
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions tests/clv/test_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,8 +86,7 @@ def fitted_gg(test_summary_data) -> GammaGammaModel:
# TODO: Consolidate this fixture into the tests requiring it?
@pytest.fixture()
def df_cum_transactions():
url_cdnow = "https://raw.githubusercontent.com/pymc-labs/pymc-marketing/main/data/cdnow_transactions.csv"
cdnow_transactions = pd.read_csv(url_cdnow)
cdnow_transactions = pd.read_csv("data/cdnow_transactions.csv")

rfm_data = rfm_summary(
cdnow_transactions,
Expand Down

0 comments on commit 6f12316

Please sign in to comment.