Skip to content

added more tests to prophet_forecast#264

Merged
jaredsnyder merged 32 commits into
mainfrom
kpi_forecast_add_more_prophet_tests
Aug 14, 2024
Merged

added more tests to prophet_forecast#264
jaredsnyder merged 32 commits into
mainfrom
kpi_forecast_add_more_prophet_tests

Conversation

@jaredsnyder
Copy link
Copy Markdown
Contributor

This PR adds unit tests for the predict, _predict, _fit, fit, and summarize methods for ProphetForecast to make it easier to refactor this code

Checklist for reviewer:

  • Commits should reference a bug or github issue, if relevant (if a bug is
    referenced, the pull request should include the bug number in the title)
  • Scan the PR and verify that no changes (particularly to
    .circleci/config.yml) will cause environment variables (particularly
    credentials) to be exposed in test logs
  • Ensure the container image will be using permissions granted to
    telemetry-airflow
    responsibly.

@jaredsnyder jaredsnyder requested a review from bochocki August 9, 2024 18:23
Copy link
Copy Markdown
Contributor

@bochocki bochocki left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This looks good! Nice work adding so many tests; hopefully this makes the updates more robust. I made a few small comments, not blocking, but I think some light touch-ups could make the code more readable.

Once this is merged, let's make sure to re-trigger the KPI Forecast Airflow task to make sure that it completes, and then verify that the dashboards aren't broken. Feel free to ping me for that step and I can validate Airflow + dashboards if you like.

Comment thread jobs/kpi-forecasting/kpi_forecasting/models/base_forecast.py Outdated
Comment thread jobs/kpi-forecasting/kpi_forecasting/models/prophet_forecast.py
Comment thread jobs/kpi-forecasting/kpi_forecasting/tests/test_prophet_forecast.py Outdated
Comment thread jobs/kpi-forecasting/kpi_forecasting/tests/test_prophet_forecast.py Outdated
Comment thread jobs/kpi-forecasting/kpi_forecasting/tests/test_prophet_forecast.py Outdated
Copy link
Copy Markdown
Contributor

@bochocki bochocki left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

👏

Base automatically changed from kpi_forecasting_funnel_unit_tests to main August 14, 2024 14:11
@jaredsnyder jaredsnyder merged commit 9a2bc3a into main Aug 14, 2024
@jaredsnyder jaredsnyder deleted the kpi_forecast_add_more_prophet_tests branch August 14, 2024 14:30
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants