소스 검색

Updating remove_weekends function

"|" is not equal "or" and drops only one weekend, not two with "or".
Marat 4 년 전
부모
커밋
32c1046873
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      stocker/stocker.py

+ 1 - 1
stocker/stocker.py

@@ -341,7 +341,7 @@ class Stocker:
 
         # Find all of the weekends
         for i, date in enumerate(dataframe["ds"]):
-            if (date.weekday()) == 5 | (date.weekday() == 6):
+            if (date.weekday()) == 5 or (date.weekday() == 6):
                 weekends.append(i)
 
         # Drop the weekends