Skip to content

Commit ac68dc1

Browse files
p1utozegithub-actions
and
github-actions
authored
Fixed Pytest warnings for machine_learning/forecasting (#8958)
* updating DIRECTORY.md * Fixed pyTest Warnings --------- Co-authored-by: github-actions <${GITHUB_ACTOR}@users.noreply.github.com>
1 parent 4b7ecb6 commit ac68dc1

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

Diff for: DIRECTORY.md

+1
Original file line numberDiff line numberDiff line change
@@ -340,6 +340,7 @@
340340
* [Rod Cutting](dynamic_programming/rod_cutting.py)
341341
* [Subset Generation](dynamic_programming/subset_generation.py)
342342
* [Sum Of Subset](dynamic_programming/sum_of_subset.py)
343+
* [Tribonacci](dynamic_programming/tribonacci.py)
343344
* [Viterbi](dynamic_programming/viterbi.py)
344345
* [Word Break](dynamic_programming/word_break.py)
345346

Diff for: machine_learning/forecasting/run.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,8 @@
1111
u can just adjust it for ur own purpose
1212
"""
1313

14+
from warnings import simplefilter
15+
1416
import numpy as np
1517
import pandas as pd
1618
from sklearn.preprocessing import Normalizer
@@ -45,8 +47,10 @@ def sarimax_predictor(train_user: list, train_match: list, test_match: list) ->
4547
>>> sarimax_predictor([4,2,6,8], [3,1,2,4], [2])
4648
6.6666671111109626
4749
"""
50+
# Suppress the User Warning raised by SARIMAX due to insufficient observations
51+
simplefilter("ignore", UserWarning)
4852
order = (1, 2, 1)
49-
seasonal_order = (1, 1, 0, 7)
53+
seasonal_order = (1, 1, 1, 7)
5054
model = SARIMAX(
5155
train_user, exog=train_match, order=order, seasonal_order=seasonal_order
5256
)

0 commit comments

Comments
 (0)